温馨提示×

Matplotlib中怎么实现3D曲面图

小亿
84
2024-05-13 13:37:15
栏目: 编程语言

要实现3D曲面图,可以使用Matplotlib中的mplot3d模块。以下是一个简单的示例代码来绘制一个3D曲面图:

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

# 生成数据
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图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

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

# 添加颜色栏
fig.colorbar(surf)

# 显示图形
plt.show()

在这个示例中,我们生成了一个二维网格数据,并计算了一个曲面函数。然后创建了一个3D图形,并使用plot_surface()方法绘制了曲面图。最后添加了一个颜色栏来表示Z值的变化。运行这段代码将绘制一个简单的3D曲面图。

0