python - NumPy 用于设置行和列的索引

Question

我有一个 NumPy 零矩阵:

import numpy as np

z = np.zeros((3,3), dtype=int)  # array([[0, 0, 0],
                                #        [0, 0, 0],
                                #        [0, 0, 0]])

接下来，我有一些索引和一些值:

idx = np.array([[0, 0],
                [1, 1],
                [1, 2],
                [2, 2]])

v = np.array([1, 3, 7, 2])

是否有任何 NumPy 函数可以将索引值设置为 my 值？得到如下:

z = array([[1, 0, 0],
           [0, 3, 7],
           [0, 0, 2]])

Answer 1

我相信您想要花哨的索引

n[idx[:,0], idx[:,1]] = v

array([[1, 0, 0],
       [0, 3, 7],
       [0, 0, 2]])