Utwórz wielowymiarowy Zeros Python

Question

Potrzebuję utworzyć wielowymiarową tablicę zer.

przez dwie (D = 2) lub trzy (D = 3) wymiar, to jest proste i że używać:

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

lub

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

Jak bo wyższego D , zrobić tablicę o długości n?

Answer 1

Możesz pomnożyć krotkę (n,) przez liczbę wymaganych wymiarów. np .:

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

EDYCJA: podczas gdy powyżej nie jest źle, to po prostu nadmierne. Odpowiedź @ mgilson jest lepsza.

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