python - 使用 matplotlib 从 'list of lists' 绘制 3d 表面

Question

我四处搜索了一下，虽然我可以找到许多有用的 meshgrid 示例，但没有一个可以清楚地说明如何将列表列表中的数据转换为我所见过的任何不同方式的可接受形式谈到。

当涉及到 numpy/matplotlib 以及我所看到的建议的术语和步骤顺序时，我有点不知所措。

我找到的最接近的是 Plotting a 3d surface from a list of tuples in matplotlib

我有一个高度数据列表列表。

data=[[h1,h2,h3,h...],
     [h,h,h,h],
     [h,h,h,h],
     [h,h,h,h16]]

data[0][1]==h2

data[4][4]==h16

如何使用 matplotlib/numpy 等生成这些值的简单 3d 表面图？就像颜色值作为 z 值的颜色图一样。我可以很好地使用 imshow()，因为它直接获取列表列表。我只是不确定我需要如何将我所拥有的东西分割成 plot_surface 可能同意的东西。

Answer 1

如果你想要一个 3d 表面，你必须生成 x 和 y 坐标。如果您不关心它们是什么而只想要表面，请生成给定长度的整数网格:

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


data = np.array(data)
length = data.shape[0]
width = data.shape[1]
x, y = np.meshgrid(np.arange(length), np.arange(width))

fig = plt.figure()
ax = fig.add_subplot(1,1,1, projection='3d')
ax.plot_surface(x, y, data)
plt.show()

请引用http://matplotlib.org/mpl_toolkits/mplot3d/tutorial.html和 http://nbviewer.ipython.org/github/jrjohansson/scientific-python-lectures/blob/master/Lecture-4-Matplotlib.ipynb了解更多信息