連続変数からのダミー変数作成
連続的な変数から、0-1のダミー変数を作成する場合を考えています。
今、マンションの階数を示す変数がkaiに入っています。
この変数から、1階の場合には、1、それ以外は0とするダミー変数 floor_1 を作成したい
と考えています。
現在は、for文とif文の組合わせで作成しているのですが、
うまくできたり、できなかったりします。
下記の場合ですと、エラーが出ています。
よりスマートな方法があれば、ご教示頂けますとありがたいです。
よろしくお願いします。
import pandas as pd
import numpy as np
import re
df = pd.DataFrame(
{'kai': ['2', '-1', '1', '20']},
index=[1, 2, 3,4])
# 1階ダミー
floor_1_ser = np.zeros((len(df),1))
for i in range(len(df)):
if df['kai'][i] == 1:
floor_1_ser[i] = 1
df["floor_1"] = floor_1_ser
df['floor_1'].describe()
del(floor_1_ser)