在python的numpy库中创建一个全零的4D矩阵

Question

我正在尝试使用以下代码在Python中创建一个四维矩阵：

import numpy as np
rho=np.zeros(2,2,2,2)

但是我遇到了以下错误；

    rho=np.zeros(2,2,2,2)
TypeError: function takes at most 3 arguments (4 given)

这在matlab中似乎可行，但在这里不行。任何帮助将不胜感激，谢谢！

- Joe Bowen

Answer 1

不要传递4个参数，而是传递一个由四个元素组成的元组作为一个参数：

>>> rho=np.zeros((2,2,2,2))
>>> rho
array([[[[ 0.,  0.],
         [ 0.,  0.]],

        [[ 0.,  0.],
         [ 0.,  0.]]],


       [[[ 0.,  0.],
         [ 0.,  0.]],

        [[ 0.,  0.],
         [ 0.,  0.]]]])
>>> rho.shape
(2, 2, 2, 2)

call signature 是 zeros(shape, dtype=float, order='C')，因此它试图将前两个解释为形状，第二个两个解释为类型，第三个两个解释为存储顺序，然后不知道如何处理最后两个。

- DSM