add a binary column based on condition

Question

I have a dataframe "veh_contract2_df" like this :

FUEL_CODE   FUEL_TYPE 
1           MARGE+PLUS
10          DIESEL

I would like to add a column "hybrid" which should contain "Y" if there is "+" in FUEL_TYPE and "N" if not.

sub_str = "+"
if(veh_contract2_df.loc[(veh_contract2_df['FUEL_TYPE'].find(sub_str)==-1)]):
    veh_contract2_df['HYBRIDE'] = "Y"
    else:
    veh_contract2_df['HYBRIDE'] ="N"

But i got this error : SyntaxError: invalid syntax (at line 71)

Any idea please? thanks

try searching for np.where. Will get you what you need.

sophocles
– sophocles

2021-04-08 15:08:26 +00:00
Commented Apr 8, 2021 at 15:08 — sophocles
– sophocles, Commented Apr 8, 2021 at 15:08

Mayank Porwal · Accepted Answer · 2021-04-08 15:08:32Z

6

Use numpy.where:

In [1923]: import numpy as np

In [1924]: df['hybrid'] = np.where(df.FUEL_TYPE.str.contains('+', regex=False), 'Y', 'N')

In [1925]: df
Out[1925]: 
   FUEL_CODE   FUEL_TYPE hybrid
0          1  MARGE+PLUS      Y
1         10      DIESEL      N

answered Apr 8, 2021 at 15:08

Mayank Porwal

34.2k9 gold badges45 silver badges65 bronze badges

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

Collectives™ on Stack Overflow

add a binary column based on condition

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related