numpy：如何使用矩陣元素作為索引？

Question

我有一個 numpy 整數矩陣，其元素表示索引。 我想創建一個大小相同的矩陣，其元素取自相應索引處的列表。

import numpy as np

matrix = np.array([[0, 1, 1], [2, 0, 1]], dtype=int)
matrix

# array([[0, 1, 1],
#        [2, 0, 1]])

values = [7, 8, 9]

values[matrix]  # gives 'ValueError: can only convert an array of size 1 to a Python scalar'

預期輸出：

    # array([[7, 8, 8],
    #        [9, 7, 8]])

我怎樣才能在 numpy 中做到這一點？

Answer 1

好的，我剛剛發現當values也是np.ndarray而不是列表時，我按預期工作。 所以這有效：

import numpy as np

matrix = np.array([[0, 1, 1], [2, 0, 1]], dtype=int)
matrix

# array([[0, 1, 1],
#        [2, 0, 1]])

values = np.array([7, 8, 9])

values[matrix]