Numpy, как сделать векторизацию присваивания на 2d массиве по индексу 1d массива

Question

Как я могу сделать следующее с помощью numpy векторизованным способом?

A=np.array([[1,2,3],[3,2,1],[5,4,6]])
index = [0,2,1]
value = [7,8,9]

for i, x in enumerate(index):
    A[i][x] = value[i]