python - 创建多维零点 Python

Question

我需要制作一个由零组成的多维数组。

对于两个 (D=2) 或三个 (D=3) 维度，这很简单，我会使用:

a = numpy.zeros(shape=(n,n)) 

或

a = numpy.zeros(shape=(n,n,n))

对于更高的 D，如何制作长度为 n 的数组？

Answer 1

您可以将元组 (n,) 乘以您想要的维数。例如:

>>> import numpy as np
>>> N=2
>>> np.zeros((N,)*1)
array([ 0.,  0.])
>>> np.zeros((N,)*2)
array([[ 0.,  0.],
       [ 0.,  0.]])
>>> np.zeros((N,)*3)
array([[[ 0.,  0.],
        [ 0.,  0.]],

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