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

Làm cách nào để in các phần tử mảng trong một phạm vi nhất định bằng Numpy?

Trong chương trình này, chúng ta phải in các phần tử của một mảng numpy trong một phạm vi nhất định. Các hàm numpy khác nhau được sử dụng là numpy.where () và numpy.logical_and ().

Thuật toán

Step 1: Define a numpy array.
Step 2: Use np.where() and np.logical_and() to find the numbers within the given range.
Step 3: Print the result.

Mã mẫu

import numpy as np

arr = np.array([1,3,5,7,10,2,4,6,8,10,36])
print("Original Array:\n",arr)

result = np.where(np.logical_and(arr>=4, arr<=20))
print(result)

Đầu ra

Original Array:
[ 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ả cung cấp vị trí chỉ mục của các phần tử thỏa mãn điều kiện trong hàm np.where ().