python - 是否有一个模仿 A[[i,j],...][...,[a,b,c]] 但不是只读的 numpy 函数？

Question

我正在尝试使用 numpy 和一些稀疏矩阵进行一些矩阵计算。为此，我想忽略矩阵中的零并只访问一些值，但我还需要覆盖它们。

import numpy as np

a=np.random.rand(5,5)
#does not change a:
a[[1,2],...][...,[1,2]]=np.array([[0,0],[0,0]])
#just changes (1,1) and (2,2)
a[[1,2],[1,2]]=np.array([0,0])

我想用零覆盖 [1,1],[1,2],[2,1],[2,2]。

Answer 1

我认为你需要这样的东西:

import numpy as np

a = np.arange(25).reshape((5, 5))
i, j = np.ix_([1, 2], [1, 2])
a[i, j] = np.zeros((2, 2))
print(a)
# [[ 0  1  2  3  4]
#  [ 5  0  0  8  9]
#  [10  0  0 13 14]
#  [15 16 17 18 19]
#  [20 21 22 23 24]]