numpy数组替换部分元素

Question

我有两个numpy浮点型数组，a和b，不用循环的方式，将数组a中和数组b相同的元素替换为nan？

Answer 1

试试这个。

>>> import numpy as np
>>> a = np.array([3.4, 2.5, 1.8, 4.7, 5.6, 2.1])
>>> b = np.array([2.5, 4.7, 5.6])
>>> np.where(np.isin(a,b), np.nan, a)
array([3.4, nan, 1.8, nan, nan, 2.1])