Kaggle：複数列の条件が一致したら１，しなければ０を返す列を作成したい

Question

Kaggleで、D,E,Fがhoge,test,fooなら1、そうでないなら0である列を作りたいです。
申し訳ございませんが、どなたか教えていただけますでしょうか。

df = pd.DataFrame({ 'A' : 1.,
                        'B' : pd.Timestamp('20130102'),
                        'C' : pd.Series(1,index=list(range(4)),dtype='float32'),
                        'D' : 'hoge',
                        'E' : pd.Categorical(["test","train","test","train"]),
                        'F' : 'foo' })

Answer 1

以下のように、まず全ての値が0である列を作成し、条件に合う行の値を1にすることで作成できます。

import pandas as pd

df = pd.DataFrame({ 'A' : 1.,
                        'B' : pd.Timestamp('20130102'),
                        'C' : pd.Series(1,index=list(range(4)),dtype='float32'),
                        'D' : 'hoge',
                        'E' : pd.Categorical(["test","train","test","train"]),
                        'F' : 'foo' })

df["out"] = 0
df["out"][(df['D'] == "hoge") & (df["E"] == "test") & (df["F"] == "foo")] = 1

print(df)

以下が実行結果です。

     A          B    C     D      E    F  out
0  1.0 2013-01-02  1.0  hoge   test  foo    1
1  1.0 2013-01-02  1.0  hoge  train  foo    0
2  1.0 2013-01-02  1.0  hoge   test  foo    1
3  1.0 2013-01-02  1.0  hoge  train  foo    0

Comments

[esklia]

ご回答まことにありがとうございます。無事目的の特徴量を作ることができました。以下のようなwarningが出てしまったのですが、これについてはご存じでしょうか？おそらくこの特殊？な記法によるものと思われます。
df["out"][(df['D'] == "hoge") & (df["E"] == "test") & (df["F"] == "foo")] = 1

A value is trying to be set on a copy of a slice from a DataFrame See the caveats in the documentation: https:

[nnkkmto]

df.loc[(df['D'] == "hoge") & (df["E"] == "test") & (df["F"] == "foo"), "out"] = 1

すみません、書き方が古かったです。こちらだとwarningでないと思います

[esklia]

ありがとうございます。warningも出なくなりました！