python - 编辑数组中的所有其他项目

Question

matrix = np.array([[0,0,0,0],[0,0,0,0],[0,0,0,0],[0,0,0,0]])
vector = np.array([0,0,0,0])

对于向量，您可以像这样编辑其他所有元素

vector[1::2] = 1

这给出

np.array([0,1,0,1])

但是；

matrix[1::2] = 1

产量

np.array([[0,0,0,0],[1,1,1,1],[0,0,0,0],[1,1,1,1]])

我想要输出

np.array([[0,1,0,1],[0,1,0,1],[0,1,0,1],[0,1,0,1]])

有一种蛮力方法可以获取数组的形状，将其展平，使用 [1::2]，然后 reshape ，但我确信我缺少一个更优雅的解决方案。

如有任何帮助，我们将不胜感激。

Answer 1

您可以使用多维索引执行类似的操作

>>> matrix
array([[0, 0, 0, 0],
       [0, 0, 0, 0],
       [0, 0, 0, 0],
       [0, 0, 0, 0]])
>>> matrix[:,1::2] = 1
>>> matrix
array([[0, 1, 0, 1],
       [0, 1, 0, 1],
       [0, 1, 0, 1],
       [0, 1, 0, 1]])