np.equal详解

BJPV • 2024-11-05 16:55 • 编程

一、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用于比较两个数组是否相等，且具有一定的误差容忍度。

原创文章，作者：BJPV，如若转载，请注明出处：https://www.506064.com/n/150002.html