python dataframe how to convert set column to list

user_1828513 0
Viewed 3494

I tried to convert a set column to list in python dataframe, but failed. Not sure what's best way to do so. Thanks.

Here is the example: I tried to create a 'c' column which convert 'b' set column to list. but 'c' is still set.

data = [{'a': [1,2,3], 'b':{11,22,33}},{'a':[2,3,4],'b':{111,222}}]
tdf = pd.DataFrame(data)
tdf['c'] = list(tdf['b'])
tdf
           a             b             c
0  [1, 2, 3]  {33, 11, 22}  {33, 11, 22}
1  [2, 3, 4]    {222, 111}    {222, 111}
dataframe set list python
2 Answers

Use apply:

tdf['c'] = tdf['b'].apply(list)

Because using list is doing to whole column not one by one.

Or do:

tdf['c'] = tdf['b'].map(list)
user_87083640

You could do:

import pandas as pd

data = [{'a': [1,2,3], 'b':{11,22,33}},{'a':[2,3,4],'b':{111,222}}]
tdf = pd.DataFrame(data)

tdf['c'] = [list(e) for e in tdf.b]

print(tdf)
user_40015920
Related