我有一个数据框:

data = {'year': [2010, 2011, 2012, 2011, 2012, 2010, 2011, 2012],
    'team': ['Bears', 'Bears', 'Bears', 'Packers', 'Packers', 'Lions', 'Lions', 'Lions'],
    'wins': ['11102', '8425', '12%', '15%', '11%', '6%', '20%', '4%'],
    'losses': ['5222', '8888', '6%', '1%', '5%', '30%', '6%', '12%'],
    }
football = pd.DataFrame(data, index=['a','b','c','d','e','f','g','ssa'], columns=['year', 'team', 'wins', 'losses'])


我也有一个清单:

fixed_cats = ['d','g','ssa']


我想检查fixed_cats列表中的项目是否在df索引的底部找到。

这是我失败的尝试:

football.loc[football.index[-len(fixed_cats):].isin(fixed_cats)]


由于某种原因,它将返回带有索引['b','c']的df。

预期产量:

索引为“ g”和“ ssa”的df

最佳答案

首次尝试看到['b','c']的原因是,内部isin返回的是[False, True, True]的布尔索引,您将从一开始就将其应用于df,则需要再次将其重新应用于最后3行:

In [21]:

fixed_cats = ['d','g','ssa']
football[-len(fixed_cats):][football.index[-len(fixed_cats):].isin(fixed_cats)]
Out[21]:
     year   team wins losses
g    2011  Lions  20%     6%
ssa  2012  Lions   4%    12%

In [22]:

football.index[-len(fixed_cats):].isin(fixed_cats)
Out[22]:
array([False,  True,  True], dtype=bool)


因此,上述布尔索引需要应用到最后3行,而不是整个df,这就是您要做的

10-06 10:25
查看更多