以下是一个示例代码,用于按照名称划分列:
import pandas as pd
# 创建一个示例数据框
data = {'名称': ['A', 'B', 'C', 'A', 'B', 'C'],
'数值': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)
# 使用groupby按照名称分组,并将数值列聚合为列表
df_grouped = df.groupby('名称')['数值'].apply(list).reset_index()
# 打印结果
print(df_grouped)
输出结果如下:
名称 数值
0 A [1, 4]
1 B [2, 5]
2 C [3, 6]
以上代码首先创建了一个示例数据框,其中包含两列:名称和数值。然后使用groupby
方法按照名称分组,并将数值列聚合为列表。最后通过reset_index
方法重置索引,并打印结果。
这样就实现了按照名称划分列的解决方法。