在数据处理中,我们经常会遇到需要将Pandas DataFrame的某列转换为列表的情况。如果你还在纠结这需要耗费精力去查阅资料,那么你错了!本文将教会你3秒内学会3种Pandas列转换为列表的方法。
1、 .values.tolist()
这是最简单直接的方法。举个例子,如果你有一个DataFrame如下:
col1 col2 col3
0 1 4 7
1 2 5 8
2 3 6 9
你可以这样将col1列转换为列表:
col1_list = df['col1'].values.tolist()
结果会是:[1, 2, 3]
2、.to_list()
这个方法效果与上面.values.tolist()一致,示例如下:
col1_list = df['col1'].to_list()
结果也是:[1, 2, 3]
3、列表推导式
你也可以使用列表推导式来转换,如下:
col1_list = [x for x in df['col1']]
结果同样是:[1, 2, 3]
总结一下,在Pandas中将DataFrame的列转换为列表,常用的有3种方法:
- .values.tolist()
- .to_list()
- 列表推导式
是不是3秒都不到,你就完全掌握了3种Pandas列转换为列表的方法啦!这些方法都简单直接,选择一个你最喜欢的就好。