如何使用NumPy在字符串数组以后缀结束的地方返回一个布尔数组的真值

  • Post category:Python

使用 NumPy 在字符串数组以后缀结束的地方返回一个布尔数组的真值可以通过以下步骤实现:

  1. 导入 NumPy 模块:
import numpy as np
  1. 创建一个字符串数组:
arr = np.array(['apple', 'banana', 'mango', 'orange'])
  1. 使用 np.char.endswith() 函数来检查字符串数组中每个元素是否以指定后缀结尾:
result = np.char.endswith(arr, 'go')
print(result)

输出结果为:

[False False  True  True]

上面的代码中,np.char.endswith() 函数接受两个参数,第一个参数是要检查的字符串数组,第二个参数是要检查的后缀,函数会返回与输入数组相同形状的布尔数组,表示每个元素是否以指定后缀结尾。

下面是一个完整的示例代码:

import numpy as np

# 创建一个字符串数组
arr = np.array(['apple', 'banana', 'mango', 'orange'])

# 检查每个元素是否以指定后缀结尾
result = np.char.endswith(arr, 'go')

# 输出结果
print(result)

输出结果为:

[False False  True  True]

另外,如果要检查多个后缀,可以将后缀作为元组传递给 np.char.endswith() 函数,例如:

import numpy as np

# 创建一个字符串数组
arr = np.array(['apple', 'banana', 'mango', 'orange'])

# 检查每个元素是否以指定后缀结尾
result = np.char.endswith(arr, ('e', 'go'))

# 输出结果
print(result)

输出结果为:

[ True False  True  True]