python之同时翻转 Numpy 数组中的两个值

如果我有一个 numpy 数组，例如:

[0,1,0,2,2] 

我想同时翻转列表中的 0 和 2(以获得 [2,1,2,0,0])，最好的方法是什么？

请您参考如下方法:

这是条件在 numpy 中的直接应用。

def switchvals(arr, val1, val2): 
    mask1 = arr == val1 
    mask2 = arr == val2 
    arr[mask1] = val2 
    arr[mask2] = val1