効率的な方法でPythonのダミー変数を作成する

Question

ダミー変数のベクトルを作成したい（Oか1しか取れない）。私は、次のことをやっている：

data = ['one','two','three','four','six'] 
variables = ['two','five','ten'] 

を、私は、次の2つの方法を得た：

dummy=[] 
for variable in variables: 
    if variable in data: 
     dummy.append(1) 
    else: 
     dummy.append(0) 

やリストの内包と：

dummy = [1 if variable in data else 0 for variable in variables] 

結果はOKです：

>>> [1,0,0] 

機能を組み込んでいますか？すごい仕事？変数が何千もの場合はちょっと遅いです。

編集は：time.time()を使用しての結果： 私は、次のデータを使用しています：

data = ['one','two','three','four','six']*100 
variables = ['two','five','ten']*100000 

• ループ（私の例から）：2.11秒
• リスト内包：1.55秒
• リスト理解度（変数は集合の型である）：0.0004992秒
• ピーターの例：0.0004999秒
• falsetrueから例：0.000502秒

Answer 1

あなたはsetにdataを変換する場合は、ルックアップが速くなります。

また、またはFalseの場合は、ブール値を整数に変換して1または0を得ることもできます。

>>> int(True) 
1 

あなたはループを通ってセットを毎回作成したセーブように、各変数のデータのセットに__contains__を呼び出すことができます。

あなたは一緒にすべてのこれらをマップすることができます。

dummy = list(map(int, map(set(data).__contains__, variables))) 

を編集：私はワンライナーを好きなように

多く、私はそれは、リストの内包表記を使用する方が読みやすいと思います。

リストの理解度にsetを作成すると、variableごとに再作成されます。だから我々は、二行必要があります。

search = set(data) 
dummy = [int(variable in search) for variable in variables] 

Answer 2

• 使用setからitem in setをあなたは1または0を取得するint型（ブール値）を使用することができますO（1）/ item in listテイクO（n）の
• を取ります。（代わりに、条件式）、代わりlist` `の、速くなる` set`を使用

---

>>> data = ['one','two','three','four','six'] 
>>> variables = ['two','five','ten'] 
>>> xs = set(data) 
>>> [int(x in xs) for x in variables] 
[1, 0, 0]