要繪制三維散點圖和曲面圖,可以使用Matplotlib中的mplot3d模塊。下面分別介紹如何繪制三維散點圖和曲面圖:
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
# 生成隨機數據
x = np.random.rand(100)
y = np.random.rand(100)
z = np.random.rand(100)
# 創建一個三維坐標軸
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 繪制散點圖
ax.scatter(x, y, z)
# 設置坐標軸標簽
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
plt.show()
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
# 生成數據
X = np.linspace(-5, 5, 100)
Y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(X, Y)
Z = np.sin(np.sqrt(X**2 + Y**2))
# 創建一個三維坐標軸
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 繪制曲面圖
ax.plot_surface(X, Y, Z, cmap='viridis')
# 設置坐標軸標簽
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
plt.show()
以上代碼分別展示了如何繪制三維散點圖和曲面圖。在繪制曲面圖時,首先使用np.meshgrid()
生成網格數據,然后使用ax.plot_surface()
繪制曲面圖。