如何将标签数组转换为一个字母向量

Question

我有如下标签:array([[1., 0., 0., 0.], [1., 0., 0., 0.], [1., 0., 0., 0.], ..., [1., 0., 0., 0.], [0., 0., 1., 0.], [0., 0., 0., 1.]], dtype=float32)

如何将其转化为一个向量，例如用字母表示每个类(,一个,一个,b,c,d]

Answer 1

您可以使用np.argmax()与map():

list(map(lambda x: chr(ord('a') + x), np.argmax(arr, axis=1)))

这个输出:

['a', 'a', 'a', 'a', 'c', 'd']