Python Numpy:将索引转换为标签

Python Numpy:将索引转换为标签,python,numpy,Python,Numpy,我想将数组转换为相应索引的标签。试试这个 将numpy导入为np a=np.array([1,0,2,0,1,0],dtype='object') 标签=[“BKG”、“警报”、“坠落”] 对于范围内的i(len(标签)): a[a==i]=标签[i] 印刷品(a) 输出: array = [1, 0, 2, ..., 0, 1, 0] LABELS = ["BKG", "ALERT", "FALL"] array = [1, 0

我想将
数组
转换为相应索引的
标签。

试试这个

将numpy导入为np
a=np.array([1,0,2,0,1,0],dtype='object')
标签=[“BKG”、“警报”、“坠落”]
对于范围内的i(len(标签)):
a[a==i]=标签[i]
印刷品(a)
输出:

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

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

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

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

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

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

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

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