将dataFrame转换为列表 [英] Convert dataFrame to list

问题描述

我有一个熊猫数据框，可以将其转换为numpy数组，如下所示:

I have a pandas dataframe that I convert to numpy array as follows:

df.values

给出以下输出:

array([[2],
       [0],
       [1],
       ..., 
       [0],
       [1],
       [0]], dtype=int64)

但是我想按以下方式获取列表:

However I want to obtain the list as follows:

[0, 2, 3]

有什么想法吗?

推荐答案

也许您可以使用

Maybe you can use iloc or loc for selecting column and then tolist:

print df
   a
0  2
1  0
2  1
3  0
4  1
5  0

print df.values
[[2]
 [0]
 [1]
 [0]
 [1]
 [0]]

print df.iloc[:, 0].tolist()
[2, 0, 1, 0, 1, 0]

或者:

print df.values.tolist()
[[2L], [0L], [1L], [0L], [1L], [0L]]

print df.iloc[:, 0].values.tolist()
[2L, 0L, 1L, 0L, 1L, 0L]

print df.loc[:, 'a'].tolist()
[2, 0, 1, 0, 1, 0]

print df['a'].tolist()
[2, 0, 1, 0, 1, 0]

但是也许您需要 展平 :

But maybe you need flatten:

print df.values.flatten()
[2 0 1 0 1 0]

print df.iloc[:, 0].values.flatten()
[2 0 1 0 1 0]

这篇关于将dataFrame转换为列表的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！