假设我有一个矩阵和一些索引。
有没有一种高效的方法来实现以下操作?
a = np.array([[1, 2, 3], [4, 5, 6]])
a_indices = np.array([[0,2], [1,2]])
有没有一种高效的方法来实现以下操作?
for i in range(2):
a[i, a_indices[i]] = 100
# a: np.array([[100, 2, 100], [4, 100, 100]])