Python numpy.in1d()用法及代码示例

numpy.in1d()函数测试一维数组的每个元素是否还存在于第二个数组中，并返回与arr1相同长度的布尔数组，如果arr1的元素位于arr2中，则返回true，否则返回False。

用法： numpy.in1d(arr1, arr2, assume_unique = False, invert = False)

参数：
arr1:[数组]输入数组。
arr2:[数组]用于测试arr1的每个值的值。
assume_unique:[bool，可选]如果为True，则假定输入数组都是唯一的，这可以加快计算速度。默认值为False。
invert:[bool，可选]如果为True，则返回数组中的值将被反转。默认值为False。

Return :[ndarray，bool]值arr1 [in1d]在arr2中。

代码1：

# Python program explaining 
# numpy.in1d() function 
    
# importing numpy as geek  
import numpy as geek  
   
arr1 = geek.array([0, 1, 2, 3, 0, 4, 5]) 
arr2 = [0, 2, 5] 
   
gfg = geek.in1d(arr1, arr2) 
   
print (gfg)

输出：

[ True False True False True False True]

代码2：

# Python program explaining 
# numpy.in1d() function 
    
# importing numpy as geek  
import numpy as geek  
   
arr1 = geek.array([0, 1, 2, 3, 0, 4, 5]) 
arr2 = [0, 2, 5] 
   
gfg = geek.in1d(arr1, arr2, invert = True) 
   
print (gfg)

输出：

[False True False True False True False]

相关用法

注：本文由纯净天空筛选整理自sanjoy_62大神的英文原创作品 numpy.in1d() function in Python。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。