Tạo Zeros đa chiều Python

Question

Tôi cần tạo một dãy số không đối xứng.

Đối với hai (D = 2) hoặc ba (D = 3) kích thước, điều này rất dễ dàng và tôi muốn sử dụng:

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

hoặc

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

Làm thế nào để tôi cho cao hơn D , làm cho mảng chiều dài n?

Answer 1

Bạn có thể nhân một tuple (n,) theo số thứ nguyên bạn muốn. ví dụ .:

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

CHỈNH SỬA: trong khi ở trên không sai, chỉ là quá mức. @ mgilson của câu trả lời là tốt hơn.

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