将NumPy数组转换为Python List结构?

将NumPy数组转换为Python List结构?

如何将NumPy数组转换为Python列表(例如[[1,2,3],[4,5,6]]),并且速度相当快?



慕虎7371278
浏览 991回答 3
3回答

BIG阳

用途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在评论中指出这一点。)

慕雪6442864

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))
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python