Python | Numpy np.assert_equal() 方法

简介

np.assert_equal() 是 Numpy 库中的一个方法，用于判断两个数组（矩阵）是否相等。它的作用是检查 two inputs 是否相等，如果不相等就抛出异常 AssertionError。

语法

numpy.testing.assert_equal(actual, desired, err_msg='', verbose=True)

参数

• actual： 数组或标量。待校验的对象。
• desired： 数组或标量。期望的对象，要求和actual'相同。
• err_msg： 字符串，可选参数。错误信息。
• verbose： 布尔型，可选参数。如果为 True，则在失败时返回算法详细信息。

返回值

如果输入 actual 和 desired 相等，则不返回任何值，否则触发异常 AssertionError。

示例

# 引入NumPy库
import numpy as np

# 定义数组
arr1 = np.array([0.1, 0.2, 0.3, 0.4, 0.5])
arr2 = np.array([0.1, 0.2, 0.3, 0.4, 0.5])
arr3 = np.array([1, 2, 3, 4, 5])
arr4 = np.array([1, 2, 3, 4, 5, 6])
arr5 = np.array([0.1, 0.2, 0.3, 0.400001, 0.5])

# 判断arr1和arr2是否相等
np.testing.assert_equal(arr1, arr2)

# 判断arr1和arr3是否相等
np.testing.assert_equal(arr1, arr3)
# AssertionError: 
# Arrays are not equal

# 判断arr1和arr4是否相等
np.testing.assert_equal(arr1, arr4)
# AssertionError: 
# Arrays are not equal (shapes (5,), (6,) mismatch)

# 判断arr1和arr5是否相等
np.testing.assert_equal(arr1, arr5)
# AssertionError: 
# Arrays are not equal (mismatch 3.9999999999999996%)

总结

np.assert_equal() 方法是 Numpy 中的一个有用的检查方法，用于比较两个数组是否相等。在进行数组比较时，我们需要保证两个矩阵的维度、形状和元素值都相同。如果不相同就会抛出异常 AssertionError。