字符串數組用字列數據幀

Question

什麼是從字符串數組像這樣得到的最簡單的方法：

arr = ['abc def ghi', 'def jkl xyz', 'abc xyz', 'jkl xyz'] 

的數據幀，其中每列是一個字，每行包含0或1，具體取決於如果該字出現在字符串中。事情是這樣的：

abc def ghi jkl xyz 
0 1 1 1 0 0 
1 0 1 0 1 1 
2 1 0 0 0 1 
3 0 0 0 1 1 

編輯：這裏是我的方法，這對我來說似乎是一個很多蟒蛇循環，而不是使用內置的大熊貓功能

labels = (' ').join(arr) 
labels = labels.split() 
labels = list(set(labels)) 
labels = sorted(labels) 

df = pd.DataFrame(np.zeros((len(arr), len(labels))), columns=labels) 
cols = list(df.columns.values) 

for i in range(len(arr)): 
    for col in cols: 
     if col in arr[i]: 
      df.set_value(i, col, 1) 

Answer 1

編輯 - 減少到3條必不可少線：

import pandas as pd 

arr = ['abc def ghi', 'def jkl xyz', 'abc xyz', 'jkl xyz'] 

words = set(' '.join(arr).split()) 
rows = [ { w : int(w in e) for w in words } for e in arr ] 
df = pd.DataFrame(rows) 

print(df) 

結果：

abc def ghi jkl xyz 
0 1 1 1 0 0 
1 0 1 0 1 1 
2 1 0 0 0 1 
3 0 0 0 1 1