Create list from columns and add column to dataframe python

Question

I have this dataframe:

and I would like to create col_list and add it to the dataframe such that it looks like this:

I have tried this code but it doesn't work:

import pandas as pd
import numpy as np


d = {'col1': [0, 2, 4], 'col2': [1, 3, 5], 'col3': [2, 4, 8]}
df = pd.DataFrame(d)
print(df)

b = df[['col1', 'col2']].to_numpy()
df = pd.DataFrame({'col1': [0, 2, 4], 'col2': [1, 3, 5], 'col3': [2, 4, 8],'col':[np.c_[b]]})

Suleman · Accepted Answer · 2021-09-28 13:05:08Z

1

Try this

import pandas as pd
import numpy as np

d = {'col1': [0, 2, 4], 'col2': [1, 3, 5], 'col3': [2, 4, 8]}
df = pd.DataFrame(d)

df["col_list"] = df.values.tolist()
df

answered Sep 28, 2021 at 13:05

Suleman

3975 silver badges12 bronze badges

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

Collectives™ on Stack Overflow

Create list from columns and add column to dataframe python

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