列表成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]])