如何创建一个numpy列表列表？

我已经看过谷歌,这里已经堆栈溢出,但似乎无处可寻.

主要问题是numpy假设你的列表必须成为一个数组,但这不是我想要的.

 A = np.array([[1,2],[3,4]],dtype=object)

您已应用一些技巧来绕过此默认行为.

一种是使子列表的长度可变.它不能从这些数组生成二维数组,所以它转向对象数组：

In [43]: A=np.array([[1,2],[],[1,2,3,4]])
In [44]: A
Out[44]: array([[1, 2], [], [1, 2, 3, 4]], dtype=object)

然后,您可以将值附加到每个列表中：

In [45]: for i in A: i.append(34)
In [46]: A
Out[46]: array([[1, 2, 34], [34], [1, 2, 3, 4, 34]], dtype=object)

np.empty还会创建一个对象数组：

In [47]: A=np.empty((3,),dtype=object)
In [48]: A
Out[48]: array([None, None, None], dtype=object)

但是,您必须小心如何将元素更改为列表. np.fill很诱人,但有问题：

In [49]: A.fill([])
In [50]: A
Out[50]: array([[], [], []], dtype=object)
In [51]: for i in A: i.append(34)
In [52]: A
Out[52]: array([[34, 34, 34], [34, 34, 34], [34, 34, 34]], dtype=object)

事实证明,fill会在所有插槽中放置相同的列表,因此修改一个会修改所有其他插槽.您可以通过列表列表获得相同的问题：

In [53]: B=[[]]*3
In [54]: B
Out[54]: [[], [], []]
In [55]: for i in B: i.append(34)
In [56]: B
Out[56]: [[34, 34, 34], [34, 34, 34], [34, 34, 34]]

初始化空A的正确方法是迭代,例如,

In [65]: A=np.empty((3,),dtype=object)
In [66]: for i,v in enumerate(A): A[i]=[v,i]
In [67]: A
Out[67]: array([[None, 0], [None, 1], [None, 2]], dtype=object)
In [68]: for v in A: v.append(34)
In [69]: A
Out[69]: array([[None, 0, 34], [None, 1, 34], [None, 2, 34]], dtype=object)

从问题和评论中有点不清楚是否要附加到列表或将列表追加到数组中.我刚刚演示了附加到列表中.

有一个np.append函数,新用户经常滥用它.它不能替代列表追加.它是np.concatenate的前端.它不是就地操作;它返回一个新数组.

定义要添加的列表也很棘手：

In [72]: np.append(A,[[1,23]])
Out[72]: array([[None, 0, 34], [None, 1, 34], [None, 2, 34], 1, 23],     dtype=object)

您需要构造另一个对象数组以连接到原始对象,例如

In [76]: np.append(A,np.empty((1,),dtype=object))
Out[76]: array([[None, 0, 34], [None, 1, 34], [None, 2, 34], None], dtype=object)

在所有这些中,列表数组比列表列表更难构建,并且不易操作或更快.你必须使它成为一个列表的二维数组才能获得一些好处.

In [78]: A[:,None]
Out[78]: 
array([[[None, 0, 34]],
       [[None, 1, 34]],
       [[None, 2, 34]]], dtype=object)

您可以重塑,转置等对象数组,其中创建和操作列表列表变得更加复杂.

In [79]: A[:,None].tolist()
Out[79]: [[[None, 0, 34]], [[None, 1, 34]], [[None, 2, 34]]]

===

如https://stackoverflow.com/a/57364472/901925所示,np.frompyfunc是一个用于创建对象数组的好工具.

np.frompyfunc(list, 0, 1)(np.empty((3,2), dtype=object))