文字列からのダミー変数作成
家の特徴を示す情報を持つ変数xより、シェアハウスと、バルコニーのある物件の場合に1とする
次のような変数shareとbalconyを作成したいと考えています。
index share balcony
0 1 1
1 0 1
2 1 0
3 1 0
変数名を示すリスト var_nameと、キーワードを示すリスト var_tango
より、目的の変数を作成したいと考えています。
次のようなコードを作成しているのですが、
nameという変数が作成されるのみで、
ループに問題があるようです。
基本的なことが分かっていないのだと思いますが、
解決する方法をご教示頂きたく、よろしくお願いします。
df = pd.DataFrame(
{'x': ['バルコニーのあるシェアハウス',
'南にバルコニーあり',
'シェアハウス',
'角部屋、シェアハウス']})
var_name = ['share','balcony']
var_tango = ['シェア','バルコニー']
for name in var_name:
df["name"]=0 # dfにvar_nameの要素の変数名を作成する
for tango in var_tango:
if "tango" in str(df['x'].iat[i]): # x内にtangoの要素が入っていた場合に下記を実行
df["name"].iat[i]= 1