python - 按组中的位置然后按该组对数据框进行排序

Question

考虑数据框 df

df = pd.DataFrame(dict(
        A=list('aaaaabbbbccc'),
        B=range(12)
    ))

print(df)

    A   B
0   a   0
1   a   1
2   a   2
3   a   3
4   a   4
5   b   5
6   b   6
7   b   7
8   b   8
9   c   9
10  c  10
11  c  11

我想对数据框进行排序，如果我按 'A' 列分组，我会从每个组中提取第一个位置，然后循环返回并从每个组中获取第二个位置(如果有的话)其余的。依此类推。

我希望结果看起来像这样

    A   B
0   a   0
5   b   5
9   c   9
1   a   1
6   b   6
10  c  10
2   a   2
7   b   7
11  c  11
3   a   3
8   b   8
4   a   4

Answer 1

您可以使用 cumcount首先是 groups 中的计数值，然后是 sort_values和 reindex通过 系列 暨:

cum = df.groupby('A')['B'].cumcount().sort_values()
print (cum)
0     0
5     0
9     0
1     1
6     1
10    1
2     2
7     2
11    2
3     3
8     3
4     4
dtype: int64

print (df.reindex(cum.index))
    A   B
0   a   0
5   b   5
9   c   9
1   a   1
6   b   6
10  c  10
2   a   2
7   b   7
11  c  11
3   a   3
8   b   8
4   a   4