Pandas dataframe replace method not working when using list

Question

I want to transform this dataframe:

    col1    col2
0   [1, 0]  [0, 1]
1   [1, 1]  [1, 1]
2   [0, 1]  [1, 0]
3   [0, 1]  [1, 1]
4   [1, 1]  [0, 1]
5   [1, 0]  [1, 0]

Into this one:

    col1    col2
0   [1, 0]  [0, 1]
1        1       1
2   [0, 1]  [1, 0]
3   [0, 1]       1
4        1  [0, 1]
5   [1, 0]  [1, 0]

I tried to use replace:

df.replace([1,1], 1)

But it didn't work.

Looks like an XY problem. Maybe you could tell us how you wind up with a df of lists instead, and why you want to get columns with mixed types (list and ints) in the end? — rafaelc
– rafaelc, Commented Nov 14, 2018 at 16:29
Use tuples instead of lists. Dataframes with lists in behave in weird ways. — Woody Pride
– Woody Pride, Commented Nov 14, 2018 at 16:30

BENY · Accepted Answer · 2018-11-14 16:29:19Z

2

Using applymap

df.applymap(lambda x :  1 if x==[1,1] else x )
Out[162]: 
     col1    col2
0  [1, 0]  [0, 1]
1       1       1

answered Nov 14, 2018 at 16:29

BENY

324k22 gold badges176 silver badges250 bronze badges

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

1 Answer 1