Python Generate dummy in dataframe based on another variable

Question

I have dataframe with many variables. I would like to generate a dummy variable based on column 1, for example. If column 1's observation is NaN, then the dummy variable is filled with 0. If column 1' observation is not missing, then the dummy variable is filled with 1. Any ideas? Thanks a lot.

Use isna method and convert the result to int

Yevhen Kuzmovych
– Yevhen Kuzmovych

2021-03-15 16:17:09 +00:00
Commented Mar 15, 2021 at 16:17 — Yevhen Kuzmovych
– Yevhen Kuzmovych, Commented Mar 15, 2021 at 16:17

Kris · Accepted Answer · 2021-03-15 16:17:59Z

2

This is the easiest way:

# sample data
import pandas as pd 
import numpy as np
df = pd.DataFrame()
df['sample'] = [1,2,np.nan,4,5,np.nan]

# create dummy column
df['dummy'] = np.where(df['sample'].isna(),0,1)

answered Mar 15, 2021 at 16:17

Kris

5992 silver badges7 bronze badges

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

Collectives™ on Stack Overflow

Python Generate dummy in 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