以下是我的功能:
def draw3D(draw_tick, matrixArray):
print "Drawing tick = %d\n" % draw_tick
matrix = matrixArray[draw_tick - 450]
fig = plt.figure()
ax = fig.gca(projection='3d')
X = np.arange(-40, 40, 1)
Y = np.arange(-40, 40, 1)
X, Y = np.meshgrid(X, Y)
Z = np.matrix[Y+40][X+40]
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()
plt.close()
我想用变量x,y,z绘制3D图。
TypeError: 'type' object has no attribute '__getitem__'
此错误指向Z:
的行Z = np.matrix[Y+40][X+40]
我想将该点(矩阵)的值存储到Z
中任何人都可以帮我解决吗?
非常感谢!
更新我的问题: 我有一个包含数百个81 * 81矩阵的matrixArray。 我想在该数组中绘制一个矩阵的图。 所以我宣布:
matrix= matrixArray[draw_tick - 450]
决定特定的一个。然后我想将矩阵位置设为X& Y,并将位置的值设为Z.但我希望我的X和Y从-40到+40,这就是为什么我在两轴上加40.
答案 0 :(得分:0)
numpy.matrix是一个类(Python中的类是$window
的实例的对象),并且您尝试访问它,就好像它是嵌套数组一样。您可能希望使用type
中的值。
答案 1 :(得分:0)
从调用帮助(np.matrix)我们得到:
| Examples
| --------
| >>> a = np.matrix('1 2; 3 4')
| >>> print a
| [[1 2]
| [3 4]]
|
| >>> np.matrix([[1, 2], [3, 4]])
| matrix([[1, 2],
| [3, 4]])
|
您必须创建矩阵的实例。也许你想做:
Z = np.matrix(YOUR_ndarray_AS_ARGUMENT)