如果我有一个 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
关于python - 同时翻转 Numpy 数组中的两个值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31757047/