Matplotlib中怎么创建3D图形

lewis 1年前 (2024-03-31) 阅读数 13 #程序编程

文章标签 matplotlib

要在Matplotlib中创建3D图形，首先需要导入相关的模块，然后使用Axes3D子类来创建一个3D坐标系。下面是一个简单的例子：

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

# 创建一个3D坐标系
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 生成一些数据
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))

# 绘制3D曲面图
ax.plot_surface(X, Y, Z, cmap='viridis')

plt.show()

在这个例子中，我们首先导入了必要的模块，然后创建了一个3D坐标系。接着生成了一些数据，并使用plot_surface方法绘制了一个3D曲面图。最后调用plt.show()显示图形。您可以根据自己的需求，使用不同的数据和样式来创建各种类型的3D图形。