Cách in phần tử mảng trong khoảng cho trước bằng NumPy

Trong chương trình này, in các phần tử mảng numpy trong khoảng cho trước. Dùng numpy.where() và numpy.logical_and().

Thuật toán

Bước 1: Định nghĩa mảng numpy.
Bước 2: Dùng np.where() và np.logical_and() tìm số trong khoảng.
Bước 3: In kết quả.

Mã ví dụ

import numpy as np
arr = np.array([1, 3, 5, 7, 10, 2, 4, 6, 8, 10, 36])
print("Mảng gốc:\n", arr)
result = np.where(np.logical_and(arr >= 4, arr <= 20))
print(result)

Kết quả

Mảng gốc: [ 1  3  5  7 10  2  4  6  8 10 36]
(array([2, 3, 4, 6, 7, 8, 9], dtype=int64),)

Giải thích

Kết quả cho vị trí index của các phần tử thỏa điều kiện trong np.where().