python - 在numpy中选择一个元素

Question

a = np.arange(4).reshape(2, 2)
a
[[0, 1]
[2, 3]]
a[  a==[0, 1] ][0] = 1 

期望 a = [[1, 1], [2, 3]] 但 'a' 没有改变。这个问题的根源是什么。谢谢

Answer 1

这里的问题是您试图分配给数组的副本，而不是 View ，因此 ndarray 保持不变。

一种方法是使用 np.where :

replace_with = np.vstack([np.ones(a.shape[0]), a[:,1]])
np.where((a==[0,1]).all(1)[:,None], replace_with, a)

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

---

这里的 replace_with 是一个 ndarray ，第一列是 ones，第二列是 a 的内容，因为它看起来像那样是你想要做的:

print(replace_with)

array([[1., 1.],
       [1., 3.]])