如何将简单的列表转换为numpy数组?行是单独的子列表,每行包含子列表中的元素。
当前回答
>>> numpy.array([[1, 2], [3, 4]])
array([[1, 2], [3, 4]])
其他回答
同样,在搜索将N层嵌套列表转换为N维数组的问题后,我一无所获,所以这里是我的解决方法:
import numpy as np
new_array=np.array([[[coord for coord in xk] for xk in xj] for xj in xi], ndmin=3) #this case for N=3
>>> numpy.array([[1, 2], [3, 4]])
array([[1, 2], [3, 4]])
OP指定“行是单独的子列表,每行包含子列表中的元素”。
假设numpy的使用不被禁止(假设numpy已经被添加到OP中),使用vstack:
import numpy as np
list_of_lists= [[1, 2, 3], [4, 5, 6], [7 ,8, 9]]
array = np.vstack(list_of_lists)
# array([[1, 2, 3],
# [4, 5, 6],
# [7, 8, 9]])
或者更简单一点(在另一个答案中提到),
array = np.array(list_of_lists)
就用熊猫吧
list(pd.DataFrame(listofstuff).melt().values)
这只适用于列表的列表
如果你有一个列表的列表的列表,你可能想尝试一些沿着
lists(pd.DataFrame(listofstuff).melt().apply(pd.Series).melt().values)
其实很简单:
>>> lists = [[1, 2], [3, 4]]
>>> np.array(lists)
array([[1, 2],
[3, 4]])