如何在由一列定义的分区中进行排序，但如何将分区保留在原位置？

Question

考虑数据框 df

df = pd.DataFrame(dict(
        A=list('XXYYXXYY'),
        B=range(8, 0, -1)
    ))

print(df)

   A  B
0  X  8
1  X  7
2  Y  6
3  Y  5
4  X  4
5  X  3
6  Y  2
7  Y  1

使用'X'按列定义的组'A'，我想[8, 7, 4, 3]按预期进行排序[3, 4, 7, 8]。但是，我想将这些行保留在原处。

   A  B
5  X  3  <-- Notice all X are in same positions
4  X  4  <-- However, `[3, 4, 7, 8]` have shifted
7  Y  1
6  Y  2
1  X  7  <-- 
0  X  8  <-- 
3  Y  5
2  Y  6

Answer 1

您可以transform用来取回新的所需索引顺序，然后用于reindex重新排列DataFrame：

# Use transform to return the new ordered index values.
new_idx = df.groupby('A')['B'].transform(lambda grp: grp.sort_values().index)

# Reindex.
df = df.reindex(new_idx.rename(None))

如果需要，可以将上面的两行合并为一条长行。

结果输出：

   A  B
5  X  3
4  X  4
7  Y  1
6  Y  2
1  X  7
0  X  8
3  Y  5
2  Y  6

请注意，如果您不关心维护旧索引，则可以直接从中重新分配transform：

df['B'] = df.groupby('A')['B'].transform(lambda grp: grp.sort_values())

产生：

   A  B
0  X  3
1  X  4
2  Y  1
3  Y  2
4  X  7
5  X  8
6  Y  5
7  Y  6