python将相同ID的字符串进行合并

问题描述

"""
id   value
1      A
1      B
1      C
2      D
2      E
2      F
变为：
id  value
1   [A,B,C]
2   [D,E,F]

pandas怎么把上面的结构变为下面的形式？
"""

解决方案：

import pandas as pd
import numpy as np
data = pd.DataFrame({'id':[1,1,1,2,2,2],'value':['A','B','C','D','E','F']})

data1 = np.array(data.groupby(['id']))#按照id进行分类
# 转变成array对象之后，可以根据data1[0][1]查看其结构

id_kinds = 2#id的种类，此例比如1、2共两种
all_value = []
for j in range(2):
    value = []
    for i in data1[j][1]['value']:
        value.append(i)
    all_value.append(value)
print(all_value)

#再创建新的dataframe
new_data = pd.DataFrame({'id':[1,2],'value':all_value})
print(new_data)

原文链接：

https://blog.csdn.net/qq_32618817/article/details/80430841?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param