Pandas在groupby之后添加计数和列

import pandas as pd

# 创建示例数据
data = {'Name': ['Alice', 'Bob', 'Charlie', 'Alice', 'Bob'],
        'Age': [25, 30, 35, 25, 30],
        'City': ['New York', 'Paris', 'London', 'New York', 'Paris']}
df = pd.DataFrame(data)

# 计算每个组的计数
count = df.groupby('Name').size()
print("每个组的计数：")
print(count)

# 在groupby之后添加新的列
df['Mean_Age'] = df.groupby('Name')['Age'].transform('mean')
print("\n添加新的列后的数据：")
print(df)

每个组的计数：
Name
Alice      2
Bob        2
Charlie    1
dtype: int64

添加新的列后的数据：
      Name  Age      City  Mean_Age
0    Alice   25  New York      25.0
1      Bob   30     Paris      30.0
2  Charlie   35    London      35.0
3    Alice   25  New York      25.0
4      Bob   30     Paris      30.0