在数据处理中,我们经常会遇到需要将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种方法: 

  1. .values.tolist()
  2. .to_list() 
  3. 列表推导式

是不是3秒都不到,你就完全掌握了3种Pandas列转换为列表的方法啦!这些方法都简单直接,选择一个你最喜欢的就好。

05-07 16:23