Numpy字符串操作 - equal()函数

Numpy是一个用于数值计算的Python库，其中包含了许多有用的函数和工具。在字符串操作方面，Numpy也提供了许多实用的功能，其中之一就是equal()函数。

函数用途

equal()函数用于比较两个字符串数组是否逐元素相等。它返回一个布尔值数组，其中每个元素表示两个字符串数组中相应元素是否相等。

语法格式

numpy.char.equal(arr1, arr2)

参数说明

• arr1 ：第一个要比较的数组
• arr2 ：第二个要比较的数组

示例代码

import numpy as np

# 示例字符串数组
str_arr1 = np.array(["hello", "world", "numpy"])
str_arr2 = np.array(["world", "hello", "numpy"])

# 比较两个字符串数组，返回一个布尔值数组
result = np.char.equal(str_arr1, str_arr2)
print(result)

输出结果为：

[False False  True]

说明两个字符串数组中，第三个元素“numpy”是相等的。

注意事项

• equal()函数对数组中的每个元素执行逐个比较。
• 如果两个数组的形状不同，则它们无法进行比较，从而引发异常。
• 如果数组中的元素不是字符串类型，则它们将被强制转换为字符串。