Python Pandas combining multiple columns into single column

Question

I have a Python Pandas data frame like following:

movie        unknown action adventure animation fantasy horror romance sci-fi

Toy Story    0       1      1          0        1       0      0       1              
Golden Eye   0       1      0          0        0       0      1       0      
Four Rooms   1       0      0          0        0       0      0       0    
Get Shorty   0       0      0          1        1       0      1       0
Copy Cat     0       0      1          0        0       1      0       0

I would like to combine the movie genres into one singe column. Output would be like this:

movie       genre

Toy Story   action, adventure, fantasy, sci-fy
Golden Eye  action, romance
Four Rooms  unknown
Get Shorty  animation, fantasy, romance
Copy Cat    adventure, horror

MaxU - stand with Ukraine · Accepted Answer · 2017-05-26 15:44:45Z

2

You can do it this way:

In [171]: df['genre'] = df.iloc[:, 1:].apply(lambda x: df.iloc[:, 1:].columns[x.astype(bool)].tolist(), axis=1)

In [172]: df
Out[172]:
        movie  unknown  action  adventure  animation  fantasy  horror  romance  sci-fi                                 genre
0   Toy Story        0       1          1          0        1       0        0       1  [action, adventure, fantasy, sci-fi]
1  Golden Eye        0       1          0          0        0       0        1       0                     [action, romance]
2  Four Rooms        1       0          0          0        0       0        0       0                             [unknown]
3  Get Shorty        0       0          0          1        1       0        1       0         [animation, fantasy, romance]
4    Copy Cat        0       0          1          0        0       1        0       0                   [adventure, horror]

PS but i don't understand how can it help you and i don't see any benefit compared to "one hot encoded" matrix

edited May 26, 2017 at 15:44

answered May 26, 2017 at 15:33

MaxU - stand with Ukraine

212k37 gold badges402 silver badges436 bronze badges

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

1 Comment

mechanical_meat Over a year ago

df['genre'] = df.apply(lambda x: df.columns[x.astype(bool)].tolist()[1:], axis=1) +1 and agree that it wouldn't offer any additional benefit

Collectives™ on Stack Overflow

Python Pandas combining multiple columns into single column

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related