matplotlib 如何创建3D高度Map

sxpgvts3  于 2023-05-01  发布在  其他
关注(0)|答案(2)|浏览(121)

我有一个2D数组Z,它存储了元素位置的高度。除了使用这里的方法,我需要创建与Z大小相同的数组X和Y,还有其他更简单的方法来创建3D高度图吗?
3D表面高度图类似于第一个表面图here

lo8azlld

lo8azlld1#

下面是matplotlib的代码

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

z = np.array([[x**2 + y**2 for x in range(20)] for y in range(20)])
x, y = np.meshgrid(range(z.shape[0]), range(z.shape[1]))

# show hight map in 3d
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(x, y, z)
plt.title('z as 3d height map')
plt.show()

# show hight map in 2d
plt.figure()
plt.title('z as 2d heat map')
p = plt.imshow(z)
plt.colorbar(p)
plt.show()

在此,z:

这里是z的2D图:

sqserrrh

sqserrrh2#

即使我同意其他人的观点,meshgrids并不难,但我仍然认为Mayavi包提供了一个解决方案(检查函数surf)
from mayavi import mlab mlab.surf(Z) mlab.show()

相关问题