Create list of lists for specific columns of Pandas DataFrame

Asked Feb 06 '20 at 16:08

Active Feb 06 '20 at 16:08

Viewed 20 times

Having this:

import pandas as pd
df = pd.DataFrame([[1,2,3,8],[33,44,55,9],[555,556,222,10]])
df.columns = ['Al', 'Bandy', 'George', 'Zulu']

I get this:

I need a list of lists only from columns Bandy and George, that results exactly like this:

[[2,3], [44,55],[556, 222]]

So far I have managed with the code below and the result is a reversed dataframe, but I am quite sure there should be a better way than reversing it back:

pd.DataFrame(list([df['Bandy'], df['George']]))

asked Feb 06 '20 at 16:08

Vityata

42,633
8
55
100

1

You can use: `df[['Bandy','George']].to_numpy().tolist()` which converts the values to array and then a list – anky Feb 06 '20 at 16:09
@anky_91 - yup, it works. Consider adding it as an anwer :) – Vityata Feb 06 '20 at 16:11
No problem . However this question has been asked before hence I closed as duplicate :) – anky Feb 06 '20 at 16:20

Create list of lists for specific columns of Pandas DataFrame

0 Answers0