Tìm các phần tử trong phạm vi bằng numpy trong Python

Đôi khi trong khi xử lý dữ liệu bằng thư viện numpy, chúng tôi có thể cần lọc ra một số số nhất định trong một phạm vi cụ thể. Điều này có thể đạt được bằng cách sử dụng một số phương pháp cài sẵn có sẵn trong numpy.

Với và toán tử

Trong cách tiếp cận này, chúng ta lấy một mảng numpy sau đó áp dụng hàm logic_and cho nó. Mệnh đề where trong numpy cũng được sử dụng để áp dụng điều kiện và điều kiện. Kết quả là một mảng hiển thị vị trí của các phần tử thỏa mãn các điều kiện phạm vi bắt buộc.

import numpy as np

A = np.array([5, 9, 11, 4, 31, 27,8])

# printing initial array
print("Given Array : ", A)

# Range 6 to 15
res = np.where(np.logical_and(A >= 6, A <= 15))

# Result
print("Array with condition : ", res)

Đầu ra

Chạy đoạn mã trên cho chúng ta kết quả sau -

Given Array : [ 5 9 11 4 31 27 8]
Array with condition : (array([1, 2, 6], dtype=int32),)

Sử dụng *

Trong cách tiếp cận này, chúng tôi sử dụng toán tử *. Nó cho kết quả là các giá trị thực thay vì vị trí của các giá trị trong mảng.

import numpy as np

A = np.array([5, 9, 11, 4, 31, 27,8])

# printing initial array
print("Given Array : ", A)

# Range 6 to 15
res = A [ (A >=6) * (A <= 15)]

# Result
print("Array with condition : ", res)

Đầu ra

Chạy đoạn mã trên cho chúng ta kết quả sau -

Given Array : [ 5 9 11 4 31 27 8]
Array with condition : [ 9 11 8]