用Matplotlib画三维图片的一个实例

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

import numpy as np
from matplotlib import cm

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

u=np.linspace(-1,1,100)
x,y=np.meshgrid(u,u)
z=x**2+y**2
ax.plot_surface(x,y,z,rstride=4,cstride=4,cmap=cm.YlGnBu_r)
plt.show()

>>

相关知识，关于np.meshgrid()函数实现的功能

import numpy as np
import matplotlib.pyplot as plt
x=np.linspace(6,9,4)
y=np.linspace(1,5,5)
mx,my=np.meshgrid(x,y)
print(x)
print(y)

print(mx)
print(my)

plt.plot(mx,my,"bo")
plt.show()

>>

[6. 7. 8. 9.]

[1. 2. 3. 4. 5.]

[[6. 7. 8. 9.]
[6. 7. 8. 9.]
[6. 7. 8. 9.]
[6. 7. 8. 9.]
[6. 7. 8. 9.]]

[[1. 1. 1. 1.]
[2. 2. 2. 2.]
[3. 3. 3. 3.]
[4. 4. 4. 4.]
[5. 5. 5. 5.]]

可视化显示为

生成两个矩阵xm和ym，矩阵xm和ym可以理解为一个网格

关于Matplotlib 3D画图优秀的文章

https://blog.csdn.net/hustqb/article/details/78180337

官方文档

https://matplotlib.org/tutorials/toolkits/mplot3d.html

原文地址：https://www.cnblogs.com/vocus/p/11371290.html