python - 从字符串标签Python中提取二进制类向量

Question

我有许多标记的类别 - “A 类”到“H 类”。我想提取一个给定标记字符串的二进制向量

'Class A' -> [1, 0, 0, 0, 0, 0, 0, 0]

'Class G' -> [0, 0, 0, 0, 0, 0, 1, 0]

等等。有没有一种Python式的方法可以做到这一点？我可以用许多 if 语句来完成，但是有没有更短的方法，只需几行代码？

Answer 1

您可以使用 ord 偏移 A 的 ord 来实现此目的:

def class_string_to_vector(class_str):
    _idx = ord(class_str.strip()[-1].upper()) - ord('A')
    _vec = [0] * 8
    _vec[_idx] = 1
    return vec