简单的问题:在numpy中你如何制作一个多维数组的数组？

Question

是的,也许我应该使用普通的Python列表,但是这里有:

我想要一个9×4多维数组/矩阵(无论如何)我想要存储数组.这些数组将是1维的,长度为4096.

所以,我希望能够像这样

column = 0                                    #column to insert into
row = 7                                       #row to insert into
storageMatrix[column,row][0] = NEW_VALUE
storageMatrix[column,row][4092] = NEW_VALUE_2
etc..

我很欣赏我可以在这里做一些有点傻/不必要的事情,但它会让我更容易在我的代码中将它结构化(因为有很多这些,以及后面要做的很多分析).

谢谢!

Answer 1

请注意,要充分利用numpy的全部功能,使用3维numpy数组会更好.将3-d数组拆分为具有1-d值的2-d数组可能会使代码复杂化并迫使您使用循环而不是内置的numpy函数.

可能值得花时间重构代码以使用优质的3-d numpy数组.

但是,如果那不是一个选项,那么:

import numpy as np
storageMatrix=np.empty((4,9),dtype='object')

通过设置dtypeto 'object',我们告诉numpy允许每个元素storageMatrix成为一个任意的Python对象.

现在,您必须将numpy数组的每个元素初始化为1-d numpy数组:

storageMatrix[column,row]=np.arange(4096)

然后你可以像这样访问数组元素:

storageMatrix[column,row][0] = 1
storageMatrix[column,row][4092] = 2