groupby包含多个元素

时间:2013-04-03 11:56:16

标签: python pandas

我正在尝试在DataFrame中恢复包含多个元素的组的行(按TYPE)。

TYPE VALEUR
M1   A
M1   B
M2   A

结果应该是:

TYPE VALEUR
M1   A
M1   B

谢谢!

2 个答案:

答案 0 :(得分: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

答案 1 :(得分:0)

如果您首先将ix设置为索引,也可以使用TYPE

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