跳转至

Python 中的 numpy.in1d()函数

原文:https://www . geeksforgeeks . org/numpy-in1d-python 中的函数/

numpy.in1d() 函数测试一维数组的每个元素是否也存在于第二个数组中,并返回一个与 arr1 长度相同的布尔数组,即 arr1 的元素在 arr2 中时为真,否则为假。

语法: numpy.in1d(arr1,arr2,假定 _unique = False,invert = False)

参数: arr 1:【array _ like】输入数组。 arr 2:【array _ like】测试 arr1 每个值的值。 假设 _ 唯一:【bool,可选】如果为 True,则输入数组都假设为唯一,这样可以加快计算速度。默认值为假。 反转:【bool,可选】如果为 True,则返回数组中的值反转。默认值为假。

返回:【ndarray,bool】arr 1[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]



回到顶部