按位置选择熊猫列


100

我只是想通过整数访问命名的熊猫列。

您可以使用来按位置选择一行df.ix[3]

但是如何按整数选择一列呢?

我的数据框:

df=pandas.DataFrame({'a':np.random.rand(5), 'b':np.random.rand(5)})

更新后提出问题。
詹森·斯特林珀

在此示例中,可能没有定义列的顺序。(“ a”可以是第一列或第二列)。
user48956 '18

Answers:


151

我想到两种方法:

>>> df
          A         B         C         D
0  0.424634  1.716633  0.282734  2.086944
1 -1.325816  2.056277  2.583704 -0.776403
2  1.457809 -0.407279 -1.560583 -1.316246
3 -0.757134 -1.321025  1.325853 -2.513373
4  1.366180 -1.265185 -2.184617  0.881514
>>> df.iloc[:, 2]
0    0.282734
1    2.583704
2   -1.560583
3    1.325853
4   -2.184617
Name: C
>>> df[df.columns[2]]
0    0.282734
1    2.583704
2   -1.560583
3    1.325853
4   -2.184617
Name: C

编辑:原来的答案建议使用,df.ix[:,2]但是现在不建议使用此功能。用户应切换到df.iloc[:,2]


28
FYI df.ix现在替换为df.iloc
yosemite_k

请注意,如果有两个具有相同名称的df.iloc [:,2]方法有效,则仅返回一列,但df [df.columns [2]]方法将返回两个具有相同名称的列。
BobbyG

54

您还可以df.icol(n)用于按整数访问列。

更新:icol不推荐使用,并且可以通过以下方式实现相同的功能:

df.iloc[:, n]  # to access the column at the nth position

2
请注意,对于即将发布的版本0.11.0,不建议使用这些方法,以后的版本中可能会删除这些方法。有关如何使用iloc / iat按位置选择的信息,请参见pandas.pydata.org/pandas-docs/dev/…
Wouter Overmeire

1
不建议使用上面的链接,因为此后已重新构建了索引文档:pandas.pydata.org/pandas-docs/stable/…。到今天为止,最新版本为0.21.0,iloc仍然是按文档记录按位置访问列的方法。
iff_or

21

您可以使用基于.loc的标签或基于.iloc方法的索引来进行包括列范围在内的列切片:

In [50]: import pandas as pd

In [51]: import numpy as np

In [52]: df = pd.DataFrame(np.random.rand(4,4), columns = list('abcd'))

In [53]: df
Out[53]: 
          a         b         c         d
0  0.806811  0.187630  0.978159  0.317261
1  0.738792  0.862661  0.580592  0.010177
2  0.224633  0.342579  0.214512  0.375147
3  0.875262  0.151867  0.071244  0.893735

In [54]: df.loc[:, ["a", "b", "d"]] ### Selective columns based slicing
Out[54]: 
          a         b         d
0  0.806811  0.187630  0.317261
1  0.738792  0.862661  0.010177
2  0.224633  0.342579  0.375147
3  0.875262  0.151867  0.893735

In [55]: df.loc[:, "a":"c"] ### Selective label based column ranges slicing
Out[55]: 
          a         b         c
0  0.806811  0.187630  0.978159
1  0.738792  0.862661  0.580592
2  0.224633  0.342579  0.214512
3  0.875262  0.151867  0.071244

In [56]: df.iloc[:, 0:3] ### Selective index based column ranges slicing
Out[56]: 
          a         b         c
0  0.806811  0.187630  0.978159
1  0.738792  0.862661  0.580592
2  0.224633  0.342579  0.214512
3  0.875262  0.151867  0.071244

6

您可以通过将列索引列表传递给dataFrame.ix来访问多个列。

例如:

>>> df = pandas.DataFrame({
             'a': np.random.rand(5),
             'b': np.random.rand(5),
             'c': np.random.rand(5),
             'd': np.random.rand(5)
         })

>>> df
          a         b         c         d
0  0.705718  0.414073  0.007040  0.889579
1  0.198005  0.520747  0.827818  0.366271
2  0.974552  0.667484  0.056246  0.524306
3  0.512126  0.775926  0.837896  0.955200
4  0.793203  0.686405  0.401596  0.544421

>>> df.ix[:,[1,3]]
          b         d
0  0.414073  0.889579
1  0.520747  0.366271
2  0.667484  0.524306
3  0.775926  0.955200
4  0.686405  0.544421

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.