python - create a column with list of values from other columns

Question

assuming this is my dataframe

import pandas as pd
import numpy as np

import pandas as pd
df = pd.DataFrame({'30': [-23, 12, -12, 10, -23, 12, -32, 15, -20, 10],
                   '40': [-30, 20, -21, 15, -33, 22, -40, 25, -22, 12],
                   '50': [-40, 25, -26, 19, -39, 32, -45, 35, -32, 18],
                   '60': [-45, 34, -29, 25, -53, 67, -55, 45, -42, 19],
})

how do I get a column at the end whose values are the list of values from other columns on that row

i.e.

The first row of the new columns will contain [-23,-30,-40,-45], the 2nd row will contain [12,20,25,34] and so on.

jezrael · Accepted Answer · 2021-10-11 13:23:21Z

4

Use:

#if need all columns to lists
df['new'] = df.to_numpy().tolist()    

#if need specify columns by list
cols = [30,40,50,60]
df['new'] = df[cols].to_numpy().tolist()   
print (df)
   30  40  50  60                   new
0 -23 -30 -40 -45  [-23, -30, -40, -45]
1  12  20  25  34      [12, 20, 25, 34]
2 -12 -21 -26 -29  [-12, -21, -26, -29]
3  10  15  19  25      [10, 15, 19, 25]
4 -23 -33 -39 -53  [-23, -33, -39, -53]
5  12  22  32  67      [12, 22, 32, 67]
6 -32 -40 -45 -55  [-32, -40, -45, -55]
7  15  25  35  45      [15, 25, 35, 45]
8 -20 -22 -32 -42  [-20, -22, -32, -42]
9  10  12  18  19      [10, 12, 18, 19]

answered Oct 11, 2021 at 13:23

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

Collectives™ on Stack Overflow

python - create a column with list of values from other columns

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