pandas中如何在每个分区窗口中获得密集排序

import pandas as pd

# 创建一个示例数据集
data = {'group': ['A', 'A', 'B', 'B', 'B', 'C', 'C'],
        'value': [4, 2, 7, 1, 5, 3, 6]}
df = pd.DataFrame(data)

# 定义自定义的排序函数
def dense_rank(x):
    x['dense_rank'] = pd.Series(range(1, len(x) + 1))
    return x

# 使用groupby和apply进行分组和排序
df = df.groupby('group').apply(dense_rank)

# 打印结果
print(df)

  group  value  dense_rank
0     A      4           1
1     A      2           2
2     B      7           1
3     B      1           2
4     B      5           3
5     C      3           1
6     C      6           2