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.]]])

Answer 2

>>> sh = (10, 10, 10, 10)
>>> z1 = zeros(10000).reshape(*sh)
>>> z1.shape
(10, 10, 10, 10)

编辑：虽然上面没有错，但这只是过度。@mgilson 的答案更好。

Answer 3

In [4]: import numpy

In [5]: n = 2

In [6]: d = 4

In [7]: a = numpy.zeros(shape=[n]*d)

In [8]: a
Out[8]: 
array([[[[ 0.,  0.],
         [ 0.,  0.]],

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


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

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

Answer 4

您可以使用方括号制作多维零数组

array_4D = np.zeros([3,3,3,3])