假设我有一张看起来像这样的 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中做到这一点?
最佳答案
您可以使用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)
关于python - 如何有效地将 bool 表转换为热向量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59458163/