Python 基於另一個變量在 dataframe 中生成 dummy

Question

我有 dataframe 有很多變量。 例如，我想根據第 1 列生成一個虛擬變量。 如果第 1 列的觀察值是 NaN，則虛擬變量用 0 填充。如果第 1 列的觀察值沒有丟失，則虛擬變量用 1 填充。有什么想法嗎？ 非常感謝。

Answer 1

這是最簡單的方法：

# 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)