您的位置:

np.equal详解

一、np.equal用法

numpy库中的np.equal函数用于对比两个数组相同位置的元素是否相等,返回一个布尔值的数组(True表示相等,False表示不相等)。该函数的参数有两个:x1和x2。其中x1和x2可以是数组或标量。当x1和x2具有不同的类型时,尝试将它们转换为相同的类型。

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([1, 2, 5, 6])

print(np.equal(a,b)) #[ True  True False False]

上述例子中,数组a和数组b都有四个元素,np.equal函数对比数组a和数组b相同位置的元素,返回一个新的包含四个元素的布尔值数组。

二、np.equal()返回值

np.equal()函数返回一个与第一个参数相同的形状的布尔值数组。np.equal()函数的返回值是由两个输入数组逐位置进行比较后得到的结果。

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([1, 2, 5, 6])
c = np.array([1, 2, 3, 4])

print(np.equal(a,b)) #[ True  True False False]
print(np.equal(a,c)) #[ True  True  True  True]
print(np.equal(b,c)) #[ True  True False False]

上述代码中,np.equal函数比较a和b数组元素的值,并将比较结果“逐位置”存储到一个新的数组中返回。对于a和c,它们的元素值都一样,所以返回的数组的每个位置都是True。对于b和c,它们的元素值不完全相同,所以返回的数组中只有前两个位置是True。

三、np.equal函数相关用法

1. np.array_equal函数的用法

与np.equal不同,np.array_equal用于比较两个数组是否相等。

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([1, 2, 5, 6])
c = np.array([1, 2, 3, 4])

print(np.array_equal(a,b)) #False
print(np.array_equal(a,c)) #True

上述代码中,我们用np.array_equal比较数组a和数组b的值,返回False;比较数组a和数组c的值,返回True。

2. np.allclose函数的用法

np.allclose函数用于比较两个数组是否相等,这两个数组可以有一定的误差容忍度。np.allclose的用法为np.allclose(a,b,atol=1e-8,rtol=1e-5)。其中a、b是两个数组,atol是绝对误差容忍度(默认值是1e-8),rtol是相对误差容忍度(默认值是1e-5)

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([1.001, 2.001, 3.001, 4.001])

print(np.allclose(a,b,atol=0.01,rtol=0.01)) #True
print(np.allclose(a,b,atol=0.0001,rtol=0.01)) #False

上述代码中,我们认为a和b的误差容忍度分别是0.01和0.0001,第一次比较得到结果为True,第二次比较得到结果为False。

四、总结

np.equal函数是numpy库中用于对比两个数组相同位置的元素是否相等的函数,返回一个布尔值的数组。同时,np.array_equal和np.allclose也是对比两个数组是否相等的方法。np.array_equal用于比较两个数组是否相等,而np.allclose用于比较两个数组是否相等,且具有一定的误差容忍度。