创建列表的多维numpy数组，并为第一个元素依次分配一个数字

Question

I'm looking to create a numpy array of lists and define the first element in each list as a number in sequence.

So far I can create the numpy array of all the first elements but they are not nested within lists as I'd like.

So I have

 B=np.arange(1,10)
 Bnew = B.reshape((3,3))
 array([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])

but I want it to look like:

 array([[[1], [2], [3]],
        [[4], [5], [6]],
        [[7], [8], [9]]])

as I will be adding more numbers to each list component as I continue to modify the matrix.

Thanks!

Answer 1

To be able to append to the cells of your array you need to make it dtype=object . You can force that using the following slightly ugly hack

a = [[i] for i in range(1, 10)]
swap = a[0]
a[0] = None # <-- this prevents the array factory from converting
            #     the innermost level of lists into an array dimension
b = np.array(a)
b[0] = swap
b.shape = 3, 3

now you can for example do

b[1,1].append(2)
b
array([[[1], [2], [3]],
       [[4], [5, 2], [6]],
       [[7], [8], [9]]], dtype=object)

Answer 2

What you want is a 3-dimensional numpy array. But reshape((3, 3)) will create a 2-dimensional array, because you are providing two dimensions to it. To create what you want, you should give a 3D shape to the reshape function:

Bnew = B.reshape((3, 3, 1))