python中matplotlib三维画图
时间: 2023-07-12 18:15:01 浏览: 182
python matplotlib绘制三维图的示例
在Python中,可以使用matplotlib库来进行三维图形的绘制。下面是一个简单的示例代码:
```python
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
# 生成数据
x = np.arange(-5, 5, 0.25)
y = np.arange(-5, 5, 0.25)
x, y = np.meshgrid(x, y)
r = np.sqrt(x**2 + y**2)
z = np.sin(r)
# 绘制图形
fig = plt.figure()
ax = Axes3D(fig)
ax.plot_surface(x, y, z, rstride=1, cstride=1, cmap='rainbow')
plt.show()
```
在这个例子中,我们生成了一个x、y平面上的函数z,然后使用plot_surface函数绘制了一个三维图形。其中,rstride和cstride参数控制网格线的密度,cmap参数控制颜色映射。你可以根据自己的需求进行调整。
阅读全文