【numpy】判断一个数组元素是否在另一个数组内 numpy.isin

        numpy使用numpy.isin(element, test_elements)，可以判断一个数组元素是否在另一个数组内。详细用法请见官方文档。

        该方法依次检查element中的元素是否在test_elements里面，支持两个不同维度的数组，返回结果的维度是第一个数组。

举例：

>>> element = 2*np.arange(4).reshape((2, 2))
>>> element
array([[0, 2],
       [4, 6]])
>>> test_elements = [1, 2, 4, 8]
>>> mask = np.isin(element, test_elements)
>>> mask
array([[False,  True],
       [ True, False]])
>>> element[mask]
array([2, 4])

————————————————
来源：https://blog.csdn.net/Jinyindao243052/article/details/114753527