如何通过使用高级索引分配来直接修改掩码数组

Question

我有一个数组，我想更改某些行上的某些值。所需的行将由布尔掩码数组寻址。然后我想修改行中的一个值：

a = np.array([[0., 0.],
              [0., 0.],
              [0., 0.],
              [0., 0.],
              [0., 0.]])
mask = np.array([False,  True,  True,  True, False])
multi = np.array([0, 1, 2], dtype=np.int64)
ind = np.array([1, 0, 1], dtype=np.int64)
res = np.array([0.02238303, 0.01624808, 0.0234094])

a[mask][multi, ind] = res
print(a)           # --> [[0. 0.] [0. 0.] [0. 0.] [0. 0.] [0. 0.]]
# The desired result --> [[0. 0.] [0. 0.02238303] [0.01624808 0.] [0. 0.0234094 ] [0. 0.]]

我知道我可以通过以下方式做到这一点：

sample = a[mask]
sample[multi, ind] = res
print(sample)     # --> [[0. 0.02238303] [0.01624808 0.] [0. 0.0234094]]
a[mask] = sample
print(a)          # --> [[0. 0.] [0. 0.02238303] [0.01624808 0.] [0. 0.0234094 ] [0. 0.]]

是否可以通过在屏蔽数组上建立索引来直接完成这项工作，例如 a[mask][multi, ind] =资源？如何？