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
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
