np.isin判断数组元素在另一数组中是否存在-优快云博客

本文详细介绍了np.isin函数的使用方法，包括如何判断数组a中的元素是否出现在数组b中，以及invert参数的作用。通过具体示例展示了该函数对高维数组的支持，并对比了invert参数为True和False时的不同结果。

np.isin用法

觉得有用的话,欢迎一起讨论相互学习~Follow Me

np.isin(a,b) 用于判定a中的元素在b中是否出现过，如果出现过返回True,否则返回False,最终结果为一个形状和a一模一样的数组。
但是当参数invert被设置为True时，情况恰好相反，如果a中元素在b中没有出现则返回True,如果出现了则返回False.

import numpy as np
# 这里使用reshape是为了验证是否对高维数组适用，返回一个和a形状一样的数组
a=np.array([1,3,7]).reshape(3,1)
b=np.arange(9).reshape(3,3)
# a 中的元素是否在b中，如果在b中显示True
Np_No_invert=np.isin(a, b, invert=False)
print("Np_No_invert\n",Np_No_invert)
# a 中的元素是否在b中，如果设置了invert=True,则情况恰恰相反，即a中元素在b中则返回False
Np_invert=np.isin(a, b, invert=True)
print("Np_invert\n",Np_invert)
# Np_No_invert
#  [[ True]
#  [ True]
#  [ True]]
# Np_invert
#  [[False]
#  [False]
#  [False]]

转载于:https://www.cnblogs.com/cloud-ken/p/9927372.html