python - 如何根据另一列替换 numpy 数组中的值？

Question

假设我有以下内容:

import numpy as np

data = np.array([
     [1,2,3],
     [1,2,3],
     [1,2,3],
     [4,5,6],         
     ])

我将如何根据第 2 列中的值更改第 3 列中的值？例如，如果第 3 列 == 3，则第 2 列 = 9。

[[1,9,3],
 [1,9,3],
 [1,9,3],
 [4,5,6]]

我看过 np.any()，但我不知道如何就地改变数组。

Answer 1

您可以使用 Numpy's slicing and indexing为达到这个。获取第三列为 3 的所有行，并将每一行的第二列更改为 9:

>>> data[data[:, 2] == 3, 1] = 9
>>> data
array([[1, 9, 3],
       [1, 9, 3],
       [1, 9, 3],
       [4, 5, 6]])