python - 如何根据其他 2 列替换 numpy 数组中的值

Question

假设我有以下数组:

import numpy as np

a = np.random.randint(10, size=(3,5))

print(a)

输出如下所示:

[[4 0 3 7 3]
[6 5 7 3 3]
[6 6 4 2 1]]

如果我想选择第 0 列元素为 4 的行并将第 2 列元素替换为 6，我可以执行以下操作。

a[a[:, 0] == 4, 2] = 6

输出如下:

[[4 0 6 7 3]
[6 5 7 3 3]
[6 6 4 2 1]]

如何选择第 0 列元素为 6 AND 第 1 列元素为 5 的行，并将第 2 列元素替换为 9，以便输出如下:

[[4 0 6 7 3]
[6 5 9 3 3]
[6 6 4 2 1]]

Answer 1

您可以对条件执行按位与操作:

a[(a[:, 0] == 6) & (a[:, 1] == 5), 2] = 9