python - 在没有循环的情况下将 numpy 数组中的 1's to 0 and 0' s 更改为 1

Question

假设我有一个 numpy 数组，我想在其中将所有 1 交换为 0，将所有 0 交换为 1(该数组将具有其他值，并且 0 和 1 没有什么特别之处)。当然，我可以遍历数组并一个一个地更改值。

是否有可以推荐使用的有效方法？ np.where() 方法有这个操作的选项吗？

Answer 1

这是使用 np.where 的一种方法，并在给定值是 0 或 1 时对其进行按位 XOR:

np.where((a==0)|(a==1), a^1, a)

---

例如:

a = np.array([[0,1,2,1], [1,2,0,3]])
print(a)
array([[0, 1, 2, 1],
       [1, 2, 0, 3]])

np.where((a==0)|(a==1), a^1, a)

array([[1, 0, 2, 0],
       [0, 2, 1, 3]])