python - 如何有效地将 bool 表转换为热向量？

Question

假设我有一张看起来像这样的 table -

Movie        Action      Scifi       Drama          Romance
Abc           True       False       False            False
Def           False      False        True            False
Ghi           False      False       False            True

我想将其转换为一个热向量，以便

Abc - [1   0    0   0]'
Def - [0   0    1   0]'
Ghi - [0   0    0   1]'

已知只有一列可以为 True。

有没有有效的方法可以在Python中做到这一点？

Answer 1

您可以使用numpy来完成此操作。

import numpy as np

Abc = np.array([True,False,False,False])
Def = np.array([False,False,True,False])
Ghi = np.array([False,False,False,True])
movies = np.array([Abc, Def, Ghi])
print("Input:")
print(movies)

#casting from boolean to integer
result  = np.array(movies, dtype=np.int)

print("Output:")
print(result)