4 dimensional array of zeros in python

Question

I want to make an 4 dimensional array of zeros in python. I know how to do this for a square array but I want the lists to have different lengths.

Right now I use this:

numpy.zeros((200,)*4)

Which gives them all length 200 but I would like to have lengths 200,20,100,20 because now I have a lot of zeros in my array that I don't use

Answer 1

You can use np.full:

>>> np.full((200,20,10,20), 0)

numpy.full

Return a new array of given shape and type, filled with fill_value.

Example :

>>> np.full((1,3,2,4), 0)
array([[[[ 0.,  0.,  0.,  0.],
         [ 0.,  0.,  0.,  0.]],

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

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