2013-04-03 136 views
0

我想在DataFrame中恢復包含多個元素的組(按類型)的行。groupby包含多個元素

TYPE VALEUR 
M1 A 
M1 B 
M2 A 

的結果應該是:

TYPE VALEUR 
M1 A 
M1 B 

謝謝!

回答

3

不知道優雅的解決方案。這裏是我的愚蠢的一個:

In [149]: m = df.groupby('TYPE').size() > 1 

In [151]: df[df['TYPE'].map(m)] 
Out[151]: 
    TYPE VALEUR 
0 M1  A 
1 M1  B 
0

你也可以使用ix如果設置TYPE作爲指數第一:

df.set_index("TYPE", inplace=True) 
df.ix[df.groupby(level=0).size() > 1]