Возьмите значения из столбца данных Pandas и создайте новый столбец с ними в виде списка.

Question

У меня есть датафрейм, который выглядит так:

Record Field11 ID LesionNumber Diagnosis1<br> 1 False 1000 1 22<br> 1 False 1000 2 88<br> 1 False 1000 3 22<br> 1 False 1000 4 24

Все идентификаторы одинаковы. И такая структура повторяется для множества разных идентификаторов.

Используя все строки с одинаковым идентификатором, я хотел бы создать новый фрейм данных, который выглядит следующим образом:

Record ID LesionNumber Diagnosis1<br> 1 1000 1, 2, 3, 4 22, 88, 22, 24

Я бы хотел, чтобы LesionNumber и Diagnosis1 отображались как упорядоченные списки.

Я новичок в Пандах и информационных фреймах, поэтому моя терминология может быть неверной. Это возможно?

Answer 1

Использование agg

df.groupby(['Record','Field11','ID']).agg(lambda x : ','.join(x.astype(str))).reset_index()
Out[634]: 
   Record  Field11    ID LesionNumber   Diagnosis1
0       1    False  1000      1,2,3,4  22,88,22,24