以下是一个示例代码,可以按照ID合并行:
import pandas as pd
# 创建示例数据
data = {'ID': [1, 1, 2, 2, 3, 3],
'Name': ['John', 'Amy', 'Tom', 'Jerry', 'Lily', 'David'],
'Age': [25, 28, 30, 32, 35, 40]}
df = pd.DataFrame(data)
# 按照ID合并行
df_merged = df.groupby('ID').agg(lambda x: ', '.join(x)).reset_index()
print(df_merged)
输出结果如下:
ID Name Age
0 1 John, Amy 25, 28
1 2 Tom, Jerry 30, 32
2 3 Lily, David 35, 40
在上述示例中,我们使用groupby
函数按照ID对数据进行分组,然后使用agg
函数将每个分组内的Name和Age列合并为一个字符串,并使用逗号作为分隔符。最后使用reset_index
函数重置索引,得到合并后的DataFrame。