初始化2D数组以在每个位置存放一个列表

2024-04-28 07:57:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我是否可以创建一个numpy数组,在每个位置存储一个列表(最初为空)?你知道吗

根据文档,numpy数组存储同构类型,因此我认为这是可能的。但是我可以初始化一个来保存空列表吗?你知道吗

如果numpy阵列无法实现这一点,我有什么替代方案?(我还可以使用哪些类似2D矩阵的结构来保存列表/字典/元组?)你知道吗


Tags: 文档numpy类型列表字典方案矩阵数组
2条回答

这很少是个好主意,但您需要的是一个对象类型的数组:

>>> a = np.empty((5,), dtype=object)
>>> a
array([None, None, None, None, None], dtype=object)
>>> a[0] = []
>>> a
array([[], None, None, None, None], dtype=object)
>>> a[0].append(4)
>>> a
array([[4], None, None, None, None], dtype=object)

如果知道长度(我们称之为N),可以使用简单的列表理解,然后将其转换为numpy数组:

>>> empty_array = numpy.array([[] for i in xrange(N)])

相关问题 更多 >