python - 如何将 one-hot 编码转换为整数？

Question

我有一个形状为 (100,10) 的 numpy 数组数据集。每一行都是一个单热编码。我想将其转换为形状为 (100,) 的 nd 数组，以便将每个向量行转换为一个整数，该整数表示非零索引的索引。有没有使用 numpy 或 tensorflow 的快速方法？

Answer 1

您可以使用 numpy.argmax或 tf.argmax .示例:

import numpy as np  
a  = np.array([[0,1,0,0],[1,0,0,0],[0,0,0,1]])
print('np.argmax(a, axis=1): {0}'.format(np.argmax(a, axis=1)))

输出:

np.argmax(a, axis=1): [1 0 3]

您可能还想查看 sklearn.preprocessing.LabelBinarizer.inverse_transform .