Python Pandas : Group by one column and see the content of all columns?

Question

我有一个像这样的数据框:

org     group   count
org1      1       2
org2      1       2
org3      2       1
org4      3       3
org5      3       3
org6      3       3

这就是我想要的，“组”列中每个唯一组的一个条目:

org     group   count
org1      1       2
org3      2       1
org4      3       3

我正在使用以下分组命令，但我仍然可以看到所有行:

df.groupby('group').head()

有人知道如何获得预期的结果吗？

Answer 1

您可以在组上drop_duplicates吗？

In [172]: df.drop_duplicates('group')
Out[172]:
    org  group  count
0  org1      1      2
2  org3      2      1
3  org4      3      3

此外，df.drop_duplicates(['group', 'count']) 在这种情况下也有效。

但是，这可能不是最好的一种非常灵活的方法。 @EdChum的Answer提供灵活性方向。