pandas replace specific string with numeric value in a new column for all rows

Question

I have a data frame with a column message, and I want to create a column media such that if for index x, df.ix[x][message]=="<Media omitted>" ,then I want df.ix[x][media] = 1

for example for the dataframe:

index    message
1        hello
2        <Media omitted>
3        hello
4        <Media omitted>

I would get:

index    message          media
1        hello             0
2        <Media omitted>   1
3        hello             0
4        <Media omitted>   1

I tried to do so only by using a loop, but I'm sure there is a smarter and faster way.

jpp · Accepted Answer · 2018-03-03 15:31:30Z

1

Try this:

df['media'] = (df['message'] == '<Media omitted>').astype(int)

Explanation

df['message'] == '<Media omitted>' creates a Boolean series.
astype(int) casts the Boolean series as integer type for display purposes.

answered Mar 3, 2018 at 15:31

jpp

166k37 gold badges301 silver badges362 bronze badges

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

2 Comments

sheldonzy Over a year ago

If I want to the same for 2 different string values? like for '<Media omitted>' and thisstringtoo ? I tried with an or operator, but it didn't work.

jpp Over a year ago

Try using isin: df['media'] = (df['message'].isin({'<Media omitted>', 'thisstringtoo'}).astype(int)

jezrael · Accepted Answer · 2018-03-03 15:31:37Z

1

I think you need convert boolean mask to int by astype:

df['media'] = (df['message'] == '<Media omitted>').astype(int)
#very similar alternative
#df['media'] = df['message'].eq('<Media omitted>').astype(int)
print (df)
               message  media
index                        
1                hello      0
2      <Media omitted>      1
3                hello      0
4      <Media omitted>      1

answered Mar 3, 2018 at 15:31

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Collectives™ on Stack Overflow

pandas replace specific string with numeric value in a new column for all rows

2 Answers 2

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related