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

Tìm phần tử trong khoảng cho trước trong mảng NumPy Python


Khi xử lý dữ liệu NumPy, cần lọc phần tử nằm trong một khoảng giá trị. Có thể dùng các phương thức tích hợp.

Cách 1: logical_and() và where()

Dùng np.logical_and(A >= 6, A <= 15) kết hợp np.where() trả về chỉ mục.

import numpy as np
A = np.array([5, 9, 11, 4, 31, 27, 8])
res = np.where(np.logical_and(A >= 6, A <= 15))
print("Chỉ mục:", res)  # (array([1, 2, 6]),)

Cách 2: Toán tử * (boolean indexing)

Dùng A[(A >= 6) * (A <= 15)] trả về trực tiếp giá trị.

import numpy as np
A = np.array([5, 9, 11, 4, 31, 27, 8])
res = A[(A >= 6) * (A <= 15)]
print("Giá trị:", res)  # [9 11 8]