如何将矩阵转换为字符？

Question

classes = ['A', 'B', 'C']

my_data = [
    [2, 1, 3],
    [1, 1, 2],
    [3, 3, 3],
    [3, 1, 3],
    [3, 1, 3],
    [3, 3, 2]
]

这里，A=1，B=2，C=3。

假设，我想首先找到矩阵 my_data 每一行中的最大值，然后我想将它们转换为 classes 中的字符。

我可以在 python 中不使用循环来完成它吗？

以下源代码对我不起作用：

def prediction_to_name(pred):
    return classes[np.argmax(pred)]

Answer 1

您需要迭代数据，除非对 0:len(my_data)< 中的每个 n 重复剪切和粘贴 result.append(classes[np.argmax(my_data[n])]) /code> 这只是手动输入循环。

import numpy as np
classes = ['A', 'B', 'C']
my_data = [[2, 1, 3],
     [1, 1, 2],
     [3, 3, 3],
     [3, 1, 3],
     [3, 1, 3],
     [3, 3, 2]]

classifiedData = [classes[np.argmax(row)] for row in my_data]
print(classifiedData) # ['C', 'C', 'A', 'A', 'A', 'A']

Answer 2

您的数据索引是从一开始的，而 python 是从零开始的，因此只需减一即可使它们相等。

>>> [classes[max(row) - 1] for row in my_data]
['C', 'B', 'C', 'C', 'C', 'C']

Answer 3

这个怎么样？

print(np.array(classes)[np.max(my_data,axis=1)-1])

结果：

['C' 'B' 'C' 'C' 'C' 'C']