how to create a variable in pandas dataframe based on another variable

Question

I have an issue where I have a dataframe data with multiple columns and I want to create a variable filter in the dataframe and assign the value 1 if activation_date is null else 0.

I have written this code but this is failing to get the results, everything is getting 0 irrespective if the dates are still present.

data['filter'] = [0 if x is not None else 1 for x in data['activation_dt']]

jezrael · Accepted Answer · 2017-06-21 11:03:04Z

1

I think you need isnull for check None or NaNs and then convert True to 1 and False to 0 by astype(int):

data = pd.DataFrame({'activation_dt':[None, np.nan, 1]})
print (data)
   activation_dt
0            NaN
1            NaN
2            1.0

data['filter'] = data['activation_dt'].isnull().astype(int)
print (data)
   activation_dt  filter
0            NaN       1
1            NaN       1
2            1.0       0

answered Jun 21, 2017 at 11:03

jezrael

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

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

Collectives™ on Stack Overflow

how to create a variable in pandas dataframe based on another variable

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