获取DataFrame列作为值列表

时间:2016-03-02 12:22:07

标签: python pandas dataframe

我试图将大熊猫DataFrame的列作为值列表。

我可以使用iloc访问第一列:

df.ix[:,[0]].values

但是,它会返回一个列表数组:

>>> df3.ix[:,[1]].values
array([[  0.],
       [  0.],
       [  0.],

如何返回数字列表?

通过按名称调用列并使用tolist()

,我可以得到我想要的内容
>>> df3['D-328'].tolist()
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 15.0,

但是,当按索引调用列时,该方法不可用:

>>> df3.ix[:,[0]].tolist()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\Anaconda\lib\site-packages\pandas\core\generic.py", line 2360, in __getattr__
    (type(self).__name__, name))
AttributeError: 'DataFrame' object has no attribute 'tolist'

1 个答案:

答案 0 :(得分:9)

我认为你可以这样尝试ix

df.ix[:, 0].tolist()

正如评论中提到的DSM,您可以这样使用iloc,如果您需要按位置选择第一列:

df.iloc[:, 0].tolist()