How can I change values of data with condition in Python?

Question

I'm not sure how to explain my problem.. But I have a series that looks like this:

0 Male 
1 Female                   
2 male                                       
3 M                                        
4 m                                  
5 LGBT                                   
6 Others                           
7 Make                          
8 Male                              
9 Man                                
10 Cis Male

And I'm trying to narrow it down to:

Male
Female
Others
LGBT

I was able to narrow it down using:

LGBT = survey['Gender'].str.contains('Trans|Neuter|queer|andro|Andro|Enby|binary|trans')
survey.loc[LGBT, 'Gender'] = 'LGBT'

Others = survey['Gender'].str.contains('N|A|p')
survey.loc[Others, 'Gender'] = 'Others'

Female = survey['Gender'].str.contains('F|Wo|f|wo')
survey.loc[Female, 'Gender'] = 'Female'

For each section, but now I'm trying to have the rest of the items into 'Male', I'm not able to do so. I tried using Regex but because 'Female' also contains 'm', it just changes everything to Male...

How can I solve this problem? What kind of function should I look into? I tried eq such as below:

survey['Gender'].eq(['Male', 'Female', 'Others', 'LGBT']).map({False: 'Male'})

But it gave me an error of:

ValueError: Lengths must be equal

And I'm getting more confused.... Any advice and help will be appreciated!

Ralubrusto · Accepted Answer · 2020-12-29 20:21:22Z

1

You can use pandas.Series.isin, which is useful when you want to compare multiple values at the same time.

Male = ~survey['Gender'].isin(['Female', 'Others', 'LGBT'])
survey.loc[Male, 'Gender'] = 'Male'

Since you have already narrowed down the other cases, you can just check if the elements are equal to any of the previously set (Female, Others, LGBT) and, if not, you set them as Male.

answered Dec 29, 2020 at 20:21

Ralubrusto

1,5532 gold badges16 silver badges28 bronze badges

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

Collectives™ on Stack Overflow

How can I change values of data with condition in Python?

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