我试图从表面示例开始工作:
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
我尝试了几件事,但我得到的只是错误。
答案 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)])