在Pandas中修改MultiIndex数据帧中的重复子索引

import pandas as pd

# 创建一个具有重复子索引的MultiIndex数据帧
data = {'A': [1, 2, 3, 4], 'B': [5, 6, 7, 8]}
index = pd.MultiIndex.from_tuples([(1, 'a'), (1, 'b'), (2, 'a'), (2, 'b')])
df = pd.DataFrame(data, index=index)

# 将MultiIndex转换为普通的DataFrame
df_reset = df.reset_index()

# 检测重复的子索引
duplicated = df_reset.duplicated(subset=df_reset.columns[:-1])

# 为重复的子索引添加唯一的后缀
df_reset.loc[duplicated, df_reset.columns[:-1]] += '_' + df_reset.groupby(df_reset.columns[:-1]).cumcount().astype(str)

# 将修改后的DataFrame重新设置为MultiIndex
df_modified = df_reset.set_index(df_reset.columns[:-1])

# 打印修改后的数据帧
print(df_modified)