Matplotlib中如何使用Axes3D进行缩放？

什么是Axes3D？

在 Matplotlib 的高级模块中，Axes3D 被称为立体图形模块，它可以让数据以三维坐标系的形式进行呈现。

在 Matplotlib 中导入 Axes3D，需要用到mpl_toolkits.mplot3d 库：

from mpl_toolkits.mplot3d import Axes3D

如何进行缩放？

Matplotlib 中使用 Axes3D 进行缩放时，可以使用 set_box_aspect 方法（前提是使用了方框样式())。这种方法可以让你按照想要的比例来缩放轴。

示例代码如下：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = Axes3D(fig)

# 构造数据
x = np.linspace(-2,2,100)
y = np.linspace(-2,2,100)
x,y = np.meshgrid(x,y)
z = np.sqrt(x**2+y**2)

# 绘制3D图
ax.plot_surface(x,y,z,rstride=1,cstride=1,cmap='rainbow')
plt.show()

我们发现，这张图里的长方体没有按照我们的期望比例进行缩放。

使用 set_box_aspect 方法可以解决这一问题，示例代码如下：

fig = plt.figure()
ax = Axes3D(fig)

# 构造数据
x = np.linspace(-2,2,100)
y = np.linspace(-2,2,100)
x,y = np.meshgrid(x,y)
z = np.sqrt(x**2+y**2)

# 绘制3D图
ax.plot_surface(x,y,z,rstride=1,cstride=1,cmap='rainbow')

# 使用 set_box_aspect 方法进行缩放
ax.set_box_aspect(aspect=[1,1,0.5])
plt.show()

我们可以看到，使用 set_box_aspect 方法，我们成功地将长方体进行了缩放，这样就能够满足我们期待的显示比例需求了。