字典值根据np.where或类似的数组

Question

我有一个整数键和值的字典：

dict1 = {100:1, 101:3, 103:2}

我有一个numpy数组：

arr1 = np.array([101, 103, 101, 101, 100, 103])

我想要一个长度与arr1相同的数组，该数组具有dict中与arr1中的每个条目相对应的值。也就是说，我想要：

arr2 = np.array([3, 2, 3, 3, 1, 2]) 

我认为我需要以某种方式使用np.where，但我正在空白。

如何在python中做到这一点？

Answer 1

您可以使用列表推导并将其转换为numpy数组

arr2 = np.array([dict1[x] for x in arr1])

Answer 2

GetProbeResult()