Pandas .loc与元组列名

时间:2016-05-11 20:08:47

标签: python pandas dataframe

我目前正在使用一个使用元组列名称的panda。当尝试像普通列一样使用.loc时,元组名称会导致它出错。

测试代码如下:

import pandas as pd
import numpy as np
df1 = pd.DataFrame(np.random.randn(6,4),
                   columns=[('a','1'), ('b','2'), ('c','3'), 'nontuple'])
df1.loc[:3, 'nontuple']
df1.loc[:3, ('c','3')]

第二行按预期工作,并从0:3显示“非元组”列。第三行不起作用,而是给出错误:

  

KeyError:“[列]中没有[('c','3')]

任何想法如何解决这个问题,而不是不使用元组作为列名?

此外,我发现即使.loc没有:

,下面的代码也能正常工作
df1.ix[:3][('c','3')]

1 个答案:

答案 0 :(得分:8)

Documenation

通过元组访问,返回DF:

In [508]: df1.loc[:3, [('c', '3')]]
Out[508]:
     (c, 3)
0  1.433004
1 -0.731705
2 -1.633657
3  0.565320

通过非元组列访问,返回系列:

In [514]: df1.loc[:3, 'nontuple']
Out[514]:
0    0.783621
1    1.984459
2   -2.211271
3   -0.532457
Name: nontuple, dtype: float64

通过非元组列访问,返回DF:

In [517]: df1.loc[:3, ['nontuple']]
Out[517]:
   nontuple
0  0.783621
1  1.984459
2 -2.211271
3 -0.532457

通过它的号码访问任何列,返回系列:

In [515]: df1.iloc[:3, 2]
Out[515]:
0    1.433004
1   -0.731705
2   -1.633657
Name: (c, 3), dtype: float64

通过它的号码访问任何列,返回DF:

In [516]: df1.iloc[:3, [2]]
Out[516]:
     (c, 3)
0  1.433004
1 -0.731705
2 -1.633657

注意:注意.loc[].iloc[]之间的差异 - 它们会以不同方式过滤行!

这就像Python的切片一样:

In [531]: df1.iloc[0:2]
Out[531]:
     (a, 1)    (b, 2)    (c, 3)  nontuple
0  0.650961 -1.130000  1.433004  0.783621
1  0.073805  1.907998 -0.731705  1.984459

这包括正确的索引边界:

In [532]: df1.loc[0:2]
Out[532]:
     (a, 1)    (b, 2)    (c, 3)  nontuple
0  0.650961 -1.130000  1.433004  0.783621
1  0.073805  1.907998 -0.731705  1.984459
2 -1.511939  0.167122 -1.633657 -2.211271