Numpy：将索引转换为标签

Question

array = [1, 0, 2, ..., 0, 1, 0]

LABELS = ["BKG", "ALERT", "FALL"]

我想将array转换为相应索引的label

Answer 1

试试这个

import numpy as np
a = np.array([1,0,2,0,1,0], dtype = 'object')
labels = [ "BKG", "ALERT", "FALL" ]

for i in range(len(labels)):
        a[a == i] = labels[i]
print(a)

输出：

array(['ALERT', 'BKG', 'FALL', 'BKG', 'ALERT', 'BKG'], dtype=object)

Answer 2

一个简单的列表理解将很好：

array = [1, 0, 2, 0, 1, 0]
labels = ["BKG", "ALERT", "FALL"]

result = [labels[i] for i in array]

输出：

['ALERT', 'BKG', 'FALL', 'BKG', 'ALERT', 'BKG']