如何在pandas中对不同的行进行分组，并将类别的计数添加到新列中？

import pandas as pd

data = {'Category': ['A', 'B', 'A', 'B', 'A', 'B']}
df = pd.DataFrame(data)

df['Count'] = df.groupby('Category')['Category'].transform('size')

print(df)

  Category  Count
0        A      3
1        B      3
2        A      3
3        B      3
4        A      3
5        B      3