How to create pandas columns and fill with values according to values in another column

Question

I have this dataframe:

text           sentiment
asdasda        positive
fsdfsdfs       negative
sdfsdfs        neutral
dfsdsd         mixed

and I want this outupu:

text           positive     negative     neutral      mixed
asdasda           1            0           0           0
fsdfsdfs          0            1           0           0
sdfsdfs           0            0           1           0
dfsdsd            0            0           0           1

How can I do it?

Mahdi F. · Accepted Answer · 2022-07-23 11:18:43Z

1

You can use pandas.get_dummies but before that you need to set column "text" as index and after getting result you need to rename all columns sentiment_positive to positive , sentiment_negative to negative, ...

import pandas as pd
# df <- your_df
res = pd.get_dummies(df.set_index('text')
                     # rename column sentiment_positive to positive , 
                     # rename column sentiment_negative to negative , ...
                    ).rename(columns = lambda x: x.split('_')[1])
print(res)

          mixed  negative  neutral  positive
text                                        
asdasda       0         0        0         1
fsdfsdfs      0         1        0         0
sdfsdfs       0         0        1         0
dfsdsd        1         0        0         0

answered Jul 23, 2022 at 11:18

Mahdi F.

24.1k5 gold badges25 silver badges32 bronze badges

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

Collectives™ on Stack Overflow

How to create pandas columns and fill with values according to values in another column

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