np.where不更改给定列表的numpy数组

Question

我想有条件地填充一个给定列表的numpy数组

遍历my_listnumpy数组的每个元素并查找dup_list以查看它是否有pair值，例如，1的pair值为35。然后，将my_list中的1更改为35

my_list = np.array([1, 2, 3, 4])
dup_list = [[1, 35], [4, 31]]

my_list的理想结果：

array([35, 2, 3, 31])

我下面的代码没有改变任何东西

for dup in dup_list:
    np.where(my_list==dup[0], dup[1], my_list) 

my_list

array([1, 2, 3, 4])

Answer 1

我想这就是你想做的：

import numpy as np

my_list = np.array([1, 2, 3, 4])
dup_list = [[1, 35], [4, 31]]

for dup in dup_list:
    my_list[np.where(my_list==dup[0])] = dup[1]

print(my_list)

结果：

[35  2  3 31]