Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

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

Để trả về mảng boolean mà ở đó là True` nơi phần tử chuỗi trong mảng kết thúc bằng hậu tố, sử dụng phương thức numpy.char.endswith() trong NumPy Python. Tham số thứ nhất là mảng đầu vào. Tham số thứ hai là hậu tố. Module numpy.char cung cấp tập hợp các thao tác chuỗi vector hóa cho mảng kiểu numpy.str_.

Các bước thực hiện

Nhập thư viện cần thiết:

import numpy as np

Tạo mảng 1 chiều chuỗi:

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

Hiển thị mảng:

print("Mảng...\n", arr)

Lấy kiểu dữ liệu:

print("\nKiểu dữ liệu mảng...\n", arr.dtype)

Lấy số chiều:

print("\nSố chiều mảng...\n", arr.ndim)

Lấy hình dạng:

print("\nHình dạng mảng...\n", arr.shape)

Lấy số phần tử:

print("\nSố phần tử trong mảng...\n", arr.size)

Kiểm tra kết thúc bằng hậu tố 'E':

print("\nKết quả (endswith)...\n", np.char.endswith(arr, 'E'))

Ví dụ hoàn chỉnh

import numpy as np

# Tạo mảng 1D chuỗi
arr = np.array(['KATIE', 'JOHN', 'KATE', 'AmY', 'BRADley'])

# Hiển thị mảng
print("Mảng...\n", arr)

# Lấy kiểu dữ liệu
print("\nKiểu dữ liệu mảng...\n", arr.dtype)

# Lấy số chiều
print("\nSố chiều mảng...\n", arr.ndim)

# Lấy hình dạng
print("\nHình dạng mảng...\n", arr.shape)

# Lấy số phần tử
print("\nSố phần tử trong mảng...\n", arr.size)

# Kiểm tra kết thúc bằng hậu tố
print("\nKết quả (endswith)...\n", np.char.endswith(arr, 'E'))

Kết quả đầu ra

Mảng... ['KATIE' 'JOHN' 'KATE' 'AmY' 'BRADley']
Kiểu dữ liệu mảng...