如何更改 3D 绘图的角度 Python

在 Python 中使用 Matplotlib 进行 3D 绘图时，我们经常需要更改视角以查看数据。本文将介绍如何在 Python 中更改 3D 绘图的角度。

准备工作

使用 Matplotlib 进行 3D 绘图需要导入以下模块：

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

创建 3D 图形

创建 3D 图形的方法是使用 Axes3D 类，然后将其传递给 plt.figure() 方法。例如，以下代码创建一个空的 3D 图形：

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

更改视角

更改 3D 绘图的视角需要调用 view_init() 方法，并提供两个参数：elev 和 azim。其中，elev 控制从地平面向上的仰角，azim 控制从 x 轴逆时针旋转的角度。

例如，以下代码将视角设置为仰角为 30 度，方位角为 45 度：

ax.view_init(elev=30, azim=45)

完整示例

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

# 创建数据
theta = np.linspace(-4 * np.pi, 4 * np.pi, 100)
z = np.linspace(-2, 2, 100)
r = z**2 + 1
x = r * np.sin(theta)
y = r * np.cos(theta)

# 创建 3D 图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制数据并设置视角
ax.plot(x, y, z, label='parametric curve')
ax.view_init(elev=30, azim=45)

# 显示图形
plt.legend()
plt.show()

结论

通过调用 view_init() 方法，可以在 Python 中更改 3D 绘图的视角。这对于查看和比较数据非常有用。