gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-01 23:43:36 25 4
gpt4 key购买 nike

假设我有以下数组:

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]]

最佳答案

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

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

关于python - 如何根据其他 2 列替换 numpy 数组中的值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/64664549/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com