在 Matplotlib 中设置 3D 图的纵横比

matplotlibpythondata visualization更新于 2025/4/22 5:00:17

要在 matplotlib 中设置 3D 图的纵横比，我们可以采取以下步骤−

使用 figure() 方法创建一个新图形或激活一个现有图形。
获取当前轴，如有必要，使用 projection='3d' 创建一个。
使用 numpy 创建数据点 R、Y 和 z。
使用 R、Y 和 z 创建曲面图。
使用 set_aspect('auto') 设置纵横比。
使用 savefig() 方法保存图形。

示例

from matplotlib import pyplot as plt
from matplotlib import cm
import numpy as np
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
fig = plt.figure()
ax = fig.gca(projection='3d')
R, Y = np.meshgrid(np.arange(0, 100, 1), np.arange(0, 60, 1))
z = 0.1 * np.abs(np.sin(R / 40) * np.sin(Y / 6))
ax.plot_surface(R, Y, z, cmap=cm.rainbow, linewidth=0)
ax.set_aspect('auto')
ax.azim = -160
ax.elev = 30
fig.savefig('data.png')
plt.show()