python - 替换 Python NumPy 数组中与某些值相等的所有元素

Question

我知道如何替换 numpy 数组中大于某些值的所有元素，喜欢 array[array > 0] = 0 ，但我真的不知道如何在不使用 for 的情况下将所有 equal 的元素替换为某些值循环，如下所示可以实现我想要的，但是有什么办法不使用 for循环？

抱歉不清楚，some_values这是一个列表，例如 [7, 8, 9]

for v in some_values:
    array[array == v] = 0

Answer 1

试试 np.isin :

array[np.isin(array, some_values)] = 0