如何比较两个 NumPy 数组?
本文重点介绍使用 NumPy 对数组进行的比较。比较两个 NumPy 数组通过检查每个对应索引处的每个元素是否相同来确定它们是否等效。
方法一:
我们一般使用==
运算符来比较两个 NumPy 数组以生成新的数组对象。如果两个 NumPy 数组等效,则使用新数组对象作为 ndarray 调用 ndarray.all() 以返回 True。
import numpy as np
an_array = np.array([[1, 2], [3, 4]])
another_array = np.array([[1, 2], [3, 4]])
comparison = an_array == another_array
equal_arrays = comparison.all()
print(equal_arrays)
输出:
True
方法二:
我们还可以使用大于、小于和等于运算符进行比较。要理解,请看下面的代码。
Syntax : numpy.greater(x1, x2[, out])
Syntax : numpy.greater_equal(x1, x2[, out])
Syntax : numpy.less(x1, x2[, out])
Syntax : numpy.less_equal(x1, x2[, out])
import numpy as np
a = np.array([101, 99, 87])
b = np.array([897, 97, 111])
print("Array a: ", a)
print("Array b: ", b)
print("a > b")
print(np.greater(a, b))
print("a >= b")
print(np.greater_equal(a, b))
print("a < b")
print(np.less(a, b))
print("a <= b")
print(np.less_equal(a, b))
输出: