列表成numpy数组列表

如何将列表的简单列表转换为numpy数组? 行是单独的子列表,每行包含子列表中的元素。

>>> numpy.array([[1, 2], [3, 4]]) array([[1, 2], [3, 4]]) 

如果列表中列出的元素数量不同,Ignacio Vazquez-Abrams的答案将不起作用。 相反,有3个选项:

1)创build一个数组数组:

 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 = len(sorted(x,key=len, reverse=True)[0]) 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数组的最佳search,所以尽pipe问题已经四年,我仍然会提供以下内容:

 >>> 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] 

请注意,@巴斯蒂安的答案#1不会形成一个连续的列表,因此我添加了concatenate

无论如何…我更喜欢它的优雅使用Numpy的hstack方法。

这很简单:

 >>> lists = [[1, 2], [3, 4]] >>> np.array(lists) array([[1, 2], [3, 4]])