python - 由 pandas 分组产生的列表顺序

Question

我有一个关于 pandas.core.groupby.GroupBy.apply 的问题

考虑

df.groupby('col1')['col2'].apply(list)

订单有保证吗？例如，以下建议:

In [1]: import pandas as pd

In [2]: df = pd.DataFrame({"col1": ['a', 'a', 'b', 'b', 'b'], "col2": [1,2,3,4,5]})

In [3]: df.groupby('col1')['col2'].apply(list)
Out[3]:
col1
a       [1, 2]
b    [3, 4, 5]
Name: col2, dtype: object

In [4]: df = pd.DataFrame({"col1": ['a', 'a', 'b', 'b', 'b'], "col2": [2,1,3,4,5]})

In [5]: df.groupby('col1')['col2'].apply(list)
Out[5]:
col1
a       [2, 1]
b    [3, 4, 5]
Name: col2, dtype: object

但总是这样吗？

编辑

在两个pandas.DataFrame.groupby和 pandas.Series.groupby提到保留每个组内的顺序。但是，在上面的示例中，df.groupby('col1')['col2'] 的类型是 pandas.core.groupby.SeriesGroupBy 我不确定这同样适用于这种情况。我需要知道(即引用代码/文档)在应用 list 时保留每个组内的顺序。

Answer 1

这就是为什么我们有 sort_values

df.sort_values(['col1','col2']).groupby('col1')['col2'].apply(list)
Out[79]: 
col1
a       [1, 2]
b    [3, 4, 5]
Name: col2, dtype: object