如何每次更改 2d numpy 數組的不同元素

Question

我有一個 numpy 二維數組（矩陣）：

[[1,2,3],
 [4,5,6],
 [7,8,9]]

我想根據這個向量掩碼更改值

[0,2,1]

因此我想改變

1, 6 and 8

例如

2, 9 and 24

向量與矩陣一樣長

Answer 1

您可以索引范圍為arange的行以及列掩碼索引：

import numpy as np

a = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9],
])

rows = np.arange(len(a))
columns = [0, 2, 1]
values = [2, 9, 24]

a[rows, columns] = values

Output：

array([[ 2,  2,  3],
       [ 4,  5,  9],
       [ 7, 24,  9]])