返回一个布尔数组,其中数组中字符串元素以Python中的后缀结尾的地方为True


要在Python NumPy中返回一个布尔数组,其中数组中字符串元素以后缀结尾的地方为True,可以使用numpy.char.endswith()方法。第一个参数是输入数组。第二个参数是后缀。numpy.char模块为numpy.str_类型的数组提供了一组矢量化的字符串操作。

步骤

首先,导入所需的库:

import numpy as np

创建一个一维字符串数组:

arr = np.array(['KATIE', 'JOHN', 'KATE', 'AmY', 'BRADley'])

显示我们的数组:

print("Array...\n",arr)

获取数据类型:

print("\nArray datatype...\n",arr.dtype)

获取数组的维度:

print("\nArray Dimensions...\n",arr.ndim)

获取数组的形状:

print("\nOur Array Shape...\n",arr.shape)

获取数组的元素个数:

print("\nNumber of elements in the Array...\n",arr.size)

要返回一个布尔数组,其中数组中字符串元素以后缀结尾的地方为True,可以使用numpy.char.endswith()方法:

print("\nResult (endswith)...\n",np.char.endswith(arr, 'E'))

示例

import numpy as np

# Create a One-Dimensional array of strings
arr = np.array(['KATIE', 'JOHN', 'KATE', 'AmY', 'BRADley'])

# Displaying our array
print("Array...\n",arr)

# Get the datatype
print("\nArray datatype...\n",arr.dtype)

# Get the dimensions of the Array
print("\nArray Dimensions...\n",arr.ndim)

# Get the shape of the Array
print("\nOur Array Shape...\n",arr.shape)

# Get the number of elements of the Array
print("\nNumber of elements in the Array...\n",arr.size)

# To return a boolean array which is True where the string element in array ends with suffix, use the numpy.char.endswith() method in Python Numpy
# The first parameter is the input array
# The second parameter is the suffix
print("\nResult (endswith)...\n",np.char.endswith(arr, 'E'))

输出

Array...
['KATIE' 'JOHN' 'KATE' 'AmY' 'BRADley']

Array datatype...
<U7

Array Dimensions...
1

Our Array Shape...
(5,)

Number of elements in the Array...
5

Result (endswith)...
[ True False True False False]

更新于:2022年3月1日

129 次浏览

启动您的职业生涯

通过完成课程获得认证

开始
广告