过滤索引Pandas groupby

时间:2016-07-24 18:13:45

标签: python pandas

我有这样的框架

frame=pd.DataFrame({'Team':['USA','GER','CAN','USA','GER','CAN'],
                'MOV':[-5,2,0,0,3,4]})

我可以做一个小组来获得平均值的MOV'为每个团队

print (frame.groupby('Team')['MOV'].mean())

输出

 Team
 CAN    2.0
 GER    2.5
 USA   -2.5
 Name: MOV, dtype: float64

我想返回一个列表或一组积极的MOV'。在这种情况下,GER'和' CAN'

1 个答案:

答案 0 :(得分:1)

means = frame.groupby('Team')['MOV'].mean()

print (list(means[means > 0].index))

means是一个系列,然后您可以通过获取该系列中大于0的所有值进行过滤。然后获取该过滤系列的索引(将包含国家/地区名称)并将其打印为名单。