TypeError' type'对象没有属性' __ getitem __'在绘制3D图时

时间:2016-11-14 19:37:22

标签: python numpy plot plt

以下是我的功能:

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.

2 个答案:

答案 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)