列表到numpy数组

2024-04-20 04:25:06 发布

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


Tags: python
3条回答

由于这是Google上将列表列表转换为Numpy数组的顶级搜索,所以我将提供以下内容,尽管这个问题已经存在4年了:

>>> x = [[1, 2], [1, 2, 3], [1]]
>>> y = numpy.hstack(x)
>>> print(y)
[1 2 1 2 3 1]

当我第一次想到这样做的时候,我对自己很满意,因为它太简单了。但是,在用一个更大的列表对其进行计时之后,执行此操作实际上更快:

>>> y = numpy.concatenate([numpy.array(i) for i in x])
>>> print(y)
[1 2 1 2 3 1]

注意,@Bastiaan的答案1不构成一个连续的列表,因此我添加了concatenate

无论如何……我更喜欢hstack方法,因为它优雅地使用了Numpy。

如果你的列表包含的列表中包含的元素数目不一,那么Ignacio Vazquez Abrams的答案将不起作用。相反,至少有3种选择:

1)制作数组:

x=[[1,2],[1,2,3],[1]]
y=numpy.array([numpy.array(xi) for xi in x])
type(y)
>>><type 'numpy.ndarray'>
type(y[0])
>>><type 'numpy.ndarray'>

2)制作一个列表数组:

x=[[1,2],[1,2,3],[1]]
y=numpy.array(x)
type(y)
>>><type 'numpy.ndarray'>
type(y[0])
>>><type 'list'>

3)首先使列表长度相等:

x=[[1,2],[1,2,3],[1]]
length = max(map(len, x))
y=numpy.array([xi+[None]*(length-len(xi)) for xi in x])
y
>>>array([[1, 2, None],
>>>       [1, 2, 3],
>>>       [1, None, None]], dtype=object)
>>> numpy.array([[1, 2], [3, 4]]) 
array([[1, 2], [3, 4]])

相关问题 更多 >