曲面绘制一个二维数组

时间:2015-05-04 16:06:31

标签: python python-2.7 matplotlib plot

我试图从表面示例开始工作:

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')
X = np.arange(-1, 20, 0.25)
Y = np.arange(0, 5, 0.25)

X, Y = np.meshgrid(X, Y)


R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
        linewidth=0, antialiased=False)
ax.set_zlim(-1.01, 1.01)

ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

我有以下二维数组:

[[29607.0], [23774.0, 52621.0], [71861.0], [9540.0, 12.9], []]

每个数组对应于Y中的位置,每个值对应于X中的位置及其在Z中的对应值。例如:

对于Y = 0,我将X = 1,Z = 29607.0 对于Y = 1,我会得到X = 1,Z = 23774.0,X = 2,Z = 52621.0

我尝试了几件事,但我得到的只是错误。

1 个答案:

答案 0 :(得分:1)

我不确定这会对您遇到的任何问题有所帮助,但您可以使用类似的内容将您的值转换为更正常的格式[[x1,y1,z1],[x2,y2,z2]...

arr = [[29607.0], [23774.0, 52621.0], [71861.0], [9540.0, 12.9]]
new_arr = []
for i, sub_arr in enumerate(arr):
  for j, z in enumerate(sub_arr):
    new_arr.append([i+1, j, z])

并将x,Y,Z作为轴= 1的切片从那个numpy版本中获得。

new_arr = np.array(new_arr)
X = new_arr[:,0]

PS如果arr已经是你可以做的ndarray了:

new_arr = np.array([[j+1, i, z] for (i, j), z in np.ndenumerate(arr)])