问题:将NumPy数组转换为Python List结构?
如何将NumPy数组转换为Python列表(例如[[1,2,3],[4,5,6]]
),并且速度相当快?
回答 0
用途tolist()
:
import numpy as np
>>> np.array([[1,2,3],[4,5,6]]).tolist()
[[1, 2, 3], [4, 5, 6]]
请注意,这会将值从它们可能具有的任何numpy类型(例如np.int32或np.float32)转换为“最近兼容的Python类型”(在列表中)。如果要保留numpy数据类型,则可以在数组上调用list(),最后得到numpy标量列表。(感谢Mr_and_Mrs_D在评论中指出这一点。)
回答 1
如果numpy数组形状为2D,则numpy .tolist方法将生成嵌套列表。
如果需要平面列表,则可以使用以下方法。
import numpy as np
from itertools import chain
a = [1,2,3,4,5,6,7,8,9]
print type(a), len(a), a
npa = np.asarray(a)
print type(npa), npa.shape, "\n", npa
npa = npa.reshape((3, 3))
print type(npa), npa.shape, "\n", npa
a = list(chain.from_iterable(npa))
print type(a), len(a), a`
回答 2
tolist()
熊猫说,即使遇到嵌套数组,也可以正常工作DataFrame
;
my_list = [0,1,2,3,4,5,4,3,2,1,0]
my_dt = pd.DataFrame(my_list)
new_list = [i[0] for i in my_dt.values.tolist()]
print(type(my_list),type(my_dt),type(new_list))
回答 3
someList = [list(map(int, input().split())) for i in range(N)]
回答 4
c = np.array([[1,2,3],[4,5,6]])
list(c.flatten())
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。