将一系列已分类的数据转换为具有包含类别和新id的多索引的dataframe

import pandas as pd

data = {
    '类别': ['A', 'A', 'B', 'B', 'C'],
    '数据': [10, 20, 30, 40, 50]
}

df = pd.DataFrame(data)

df_multi_index = df.set_index(['类别', df.groupby('类别').cumcount()])

print(df_multi_index)

print(df_multi_index.loc['A'])

print(df_multi_index.loc[('A', 0)])

print(df_multi_index.loc['A', :])

print(df_multi_index.xs(0, level=1))

print(df_multi_index.groupby('类别').mean())