根据 Pandas 中的组大小对分组数据进行排序

Question

我的数据集中有两列，col1 和 col2。我想按照 col1 对数据进行分组，然后根据每组的大小对数据进行排序。也就是说，我想按大小的升序显示组。

我已经编写了用于分组和显示数据的代码如下：

 grouped_data = df.groupby('col1')
"""code for sorting comes here"""
for name,group in grouped_data:
          print (name)
          print (group)

在显示数据之前，我需要根据组大小对其进行排序，但我无法做到这一点。

原文由 krackoder 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

对于 Pandas 0.17+，使用 sort_values ：

 df.groupby('col1').size().sort_values(ascending=False)

对于 pre-0.17，您可以使用 size().order() ：

 df.groupby('col1').size().order(ascending=False)

原文由 Victor Yan 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

您可以使用 python 的 sorted ：

 In [11]: df = pd.DataFrame([[1, 2], [1, 4], [5, 6]], index=['a', 'b', 'c'], columns=['A', 'B'])

In [12]: g = df.groupby('A')

In [13]: sorted(g,  # iterates pairs of (key, corresponding subDataFrame)
                key=lambda x: len(x[1]),  # sort by number of rows (len of subDataFrame)
                reverse=True)  # reverse the sort i.e. largest first
Out[13]:
[(1,    A  B
     a  1  2
     b  1  4),
 (5,    A  B
     c  5  6)]

注意：作为迭代器 g ，迭代密钥对和相应的子帧：

 In [14]: list(g)  # happens to be the same as the above...
Out[14]:
[(1,    A  B
     a  1  2
     b  1  4,
 (5,    A  B
     c  5  6)]

原文由 Andy Hayden 发布，翻译遵循 CC BY-SA 3.0 许可协议