創建多維零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.]]]])