將熊貓列表轉換爲虛擬變量

Question

我有一個熊貓數據框，其中包含我想要轉換爲虛擬變量的變量列表。基本上，我想轉換：

這樣：

Answer 1

df = pd.DataFrame({0: [['hello', 'motto'], ['motto', 'mania']]}) 
print(df) 

       0 
0 [hello, motto] 
1 [motto, mania] 

使用str.join其次str.get_dummies

df[0].str.join('|').str.get_dummies() 

    hello mania motto 
0  1  0  1 
1  0  1  1 

Answer 2

下面是一個存儲器中保存的解決方案這是會使用稀疏矩陣和Pandas.SparseSeries：

​​

結果：

In [81]: df 
Out[81]: 
    hello mania motto 
0  1  0  1 
1  0  1  1 

In [82]: df.memory_usage() 
Out[82]: 
Index 80 
hello  8 # notice memory usage: # of ones multiplied by 8 bytes (int64) 
mania  8 
motto 16 
dtype: int64