Python | Numpy np.assert_array_less() 方法

np.assert_array_less() 是 numpy 中用于检查两个数组是否严格按元素比较逐元素小的函数。

语法

np.assert_array_less(a, b, err_msg='', verbose=True)

参数说明

• a : 第一个输入的数组。
• b : 第二个输入的数组。
• err_msg : 错误消息中显示的字符串。
• verbose : 如果为 True，则在引发异常时显示完整的错误信息。

示例

以下示例将演示 np.assert_array_less() 函数的用法：

import numpy as np

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

# 使用 np.assert_array_less() 函数检查 a 和 b 是否按元素比较逐元素小
np.assert_array_less(a, b)

上面的代码将不会引发异常，因为 a 中的每个元素都比 b 中的对应元素小。

接下来，让我们看一下以下代码示例：

import numpy as np

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

# 使用 np.assert_array_less() 函数检查 a 和 b 是否按元素比较逐元素小
np.assert_array_less(a, b)

在这种情况下，由于 b[1] 大于 a[1]，所以会引发 AssertionError 异常，其中包含错误消息。

异常

如果两个输入的数组不按元素比较逐元素小，则会引发 AssertionError 异常。

在这种情况下，错误消息将根据传递给 assert_array_less() 函数的 err_msg 参数提供的字符串显示。例如，以下代码将在引发异常时显示自定义错误消息：

import numpy as np

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

# 使用 np.assert_array_less() 函数检查 a 和 b 是否按元素比较逐元素小
np.assert_array_less(a, b, err_msg='a 中的元素必须全部小于 b 中的元素。')

输出结果如下：

AssertionError: a 中的元素必须全部小于 b 中的元素。

引用

• numpy.assert_array_less()