Computer >> Máy Tính >  >> Lập trình >> Python

Trả về một mảng boolean là True trong đó phần tử chuỗi trong mảng kết thúc bằng hậu tố trong Python

Để trả về một mảng boolean là True trong đó phần tử chuỗi trong mảng kết thúc bằng hậu tố, hãy sử dụng phương thức numpy.char.endswith () trong Python Numpy. Tham số đầu tiên là mảng đầu vào. Tham số thứ hai là hậu tố. Mô-đun numpy.char cung cấp một tập hợp các hoạt động chuỗi được vector hóa cho các mảng kiểu numpy.str_

Các bước

Đầu tiên, hãy nhập các thư viện được yêu cầu -

import numpy as np

Tạo mảng chuỗi một chiều -

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

Hiển thị mảng của chúng tôi -

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

Nhận loại dữ liệu -

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

Nhận các kích thước của Mảng -

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

Nhận hình dạng của Mảng -

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

Nhận số phần tử của Mảng -

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

Để trả về một mảng boolean là True trong đó phần tử chuỗi trong mảng kết thúc bằng hậu tố, hãy sử dụng phương thức numpy.char.endswith () -

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

Ví dụ

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'))

Đầu ra

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]