如何根据groupby操作生成的组获取pandas DataFrame的组id值的列

import pandas as pd

# 创建一个示例DataFrame
data = {'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
        'B': ['one', 'one', 'two', 'two', 'two', 'one', 'two', 'one'],
        'C': [1, 2, 3, 4, 5, 6, 7, 8]}
df = pd.DataFrame(data)

# 使用groupby操作分组，并获取组ID值的列
df['group_id'] = df.groupby(['A', 'B']).ngroup()

print(df)

     A    B  C  group_id
0  foo  one  1         0
1  bar  one  2         1
2  foo  two  3         2
3  bar  two  4         3
4  foo  two  5         2
5  bar  one  6         1
6  foo  two  7         2
7  foo  one  8         0