Take the values from Pandas dataframe column and create a new column with them as a list

Question

I've got a dataframe which looks like this:

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

All of the ID's are the same. And this kind of structure repeats for many different ID's.

Using all rows with the same ID, I'd like to create a new dataframe which looks like this:

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

I'd like to have the LesionNumber and Diagnosis1 appear as ordered lists.

I'm new to Pandas and dataframes so my terminology may be off. Is this possible?

BENY · Accepted Answer · 2018-07-06 01:56:25Z

1

Using 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

answered Jul 6, 2018 at 1:56

BENY

324k22 gold badges176 silver badges250 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Take the values from Pandas dataframe column and create a new column with them as a list

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related