要按照数据框的某一列分组,并将行合并为列,我们可以使用pandas库来完成。下面是一个示例代码:
import pandas as pd
# 创建一个示例数据框
df = pd.DataFrame({
'group': ['A', 'A', 'B', 'B', 'C'],
'value': [1, 2, 3, 4, 5]
})
# 按照'group'列分组,并将行合并为列
grouped_df = df.groupby('group')['value'].apply(list).reset_index()
print(grouped_df)
输出结果为:
group value
0 A [1, 2]
1 B [3, 4]
2 C [5]
在这个示例中,我们首先创建了一个包含两列的数据框df,其中'group'列表示分组的标识,'value'列为对应的值。然后,我们使用groupby函数按照'group'列进行分组,并使用apply函数将每个分组的'value'列转换为列表形式。最后,使用reset_index函数重置索引,以便得到最终的结果。