本文介绍了将dataFrame转换为列表的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
我有一个熊猫数据框，可以将其转换为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]
有什么想法吗?
推荐答案
也许您可以使用 iloc 或 loc 选择列，然后选择 tolist :
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转换为列表的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！