我有一个标签名称列表，我对其进行了编号并创建了一个字典:

my_list = [b'airplane', b'automobile', b'bird', b'cat', b'deer', b'dog', b'frog', b'horse', b'ship', b'truck']
label_dict = dict(enumerate(my_list))
{0: b'airplane', 1: b'automobile', 2: b'bird', 3: b'cat', 4: b'deer', 5: b'dog', 6: b'frog', 7: b'horse', 8: b'ship', 9: b'truck'}

现在我正在尝试清理并将字典值映射/应用到我的目标上，它是以独热编码的形式存在的。

y_test[0]
array([ 0.,  0.,  0.,  1.,  0.,  0.,  0.,  0.,  0.,  0.])
y_test[0].map(label_dict) should return: 'cat'

我尝试使用

(lambda key,value: value for y_test[0] == 1)

但没有找到具体的解决方案

谢谢你。

回答：

由于我们正在处理独热编码数组，可以使用argmax来获取每行中唯一1的索引。因此，使用列表作为输入 –

[my_list[i] for i in y_test.argmax(1)]

或者使用np.take来获得数组输出 –

np.take(my_list,y_test.argmax(1))

如果使用dict，并假设键是按顺序的0,1,..，我们可以这样做 –

np.take(label_dict.values(),y_test.argmax(1))

如果键不是按顺序但已排序 –

np.take(label_dict.values(), np.searchsorted(label_dict.keys(),y_test.argmax(1)))

样本运行 –

In [79]: my_list
Out[79]: ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
In [80]: y_test
Out[80]: array([[ 0.,  0.,  0.,  1.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  1.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.,  0.]])
In [81]: [my_list[i] for i in y_test.argmax(1)]
Out[81]: ['cat', 'automobile', 'ship']
In [82]: np.take(my_list,y_test.argmax(1))
Out[82]: array(['cat', 'automobile', 'ship'],
      dtype='|S10')