python - 如何以特殊方式对 Pandas 数据框进行排序

Question

给定一个 Pandas 数据框

df = pd.DataFrame({'a': [1,2,3,4,5,6,7,8],
                   'b': [0,0,1,1,2,2,3,3]})

如何沿列排序 b以这样的方式将其重新排列为 {0,1,2,3,0,1,2,3} .

IE。结果数据帧是

1   0
3   1
5   2
7   3
2   0
4   1
6   2
8   3

Answer 1

使用 cumcount 添加一列

df.assign(x=df.groupby('b').cumcount()).sort_values(['x', 'b']).drop('x', axis=1)

   a  b
0  1  0
2  3  1
4  5  2
6  7  3
1  2  0
3  4  1
5  6  2
7  8  3

Numpy 的 lexsort , iloc , 和 cumcount

df.iloc[np.lexsort([df['b'], df.groupby('b').cumcount()])]

   a  b
0  1  0
2  3  1
4  5  2
6  7  3
1  2  0
3  4  1
5  6  2
7  8  3