根据熊猫数据框中的条件删除每组的最后一行

Question

我有以下数据框：

    name    gender    count
0     A      M         3
1     A      F         2
2     A      Nan       3
3     B      NaN       2
4     C      F         4
5     D      M         5
6     D      Nan       5

我想构建一个结果数据帧df1 ，如果该组的计数大于 1，它会删除name属性组的最后一行。对于 eq- name A 出现 3 次，因此应删除包含 A 的最后一行。 B 和 C 只出现一次，因此应该保留包含它们的行。

结果数据帧df1应该是这样的：

     name    gender    count
0     A      M         3
1     A      F         2
2     B      NaN       2
3     C      F         4
4     D      M         5

请指教。

Answer 1

用

In [4598]: (df.groupby('name').apply(lambda x: x.iloc[:-1] if len(x)>1 else x)
              .reset_index(drop=True))
Out[4598]:
  name gender  count
0    A      M      3
1    A      F      2
2    B    NaN      2
3    C      F      4
4    D      M      5

Answer 2

使用groupby + head ：

g = df.groupby('name', as_index=False, group_keys=False)\
          .apply(lambda x: x.head(-1) if x.shape[0] > 1 else x)
print(g)
  name gender  count
0    A      M      3
1    A      F      2
3    B    NaN      2
4    C      F      4
5    D      M      5