Numpy Array Tìm Kiếm

Tìm kiếm mảng array trong numpy có nghĩa là tìm kiếm một hoặc một số giá trị trong mảng đã cho và trả về kết quả là các giá trị chỉ số index.

Để tìm kiếm một mảng trong numpy sử dụng phương thức where().

Ví dụ tìm phần tử có giá trị là 6 trong mảng sau:

Ví dụ

import numpy as np

arr = np.array([1,2,6,4,6,3,6])
x_search = np.where(arr == 6)
print(x_search)

Kết quả :

(array([2, 4, 6], dtype=int64),)

Kết quả của ví dụ trên trả về một tuple : (array([2, 4, 6], dtype=int64),) , nó có nghĩa là các phần tử có giá trị là 6 tại index 2, 4, 6.

Ví dụ tìm chỉ số index nơi có giá trị là chẵn (even), lẻ (odd):

Ví dụ

import numpy as np

#Ví dụ tìm chỉ số index nơi có giá trị là chẵn (even):
arr = np.array([1,2,6,4,6,3,6])
x1_search = np.where(arr%2 == 0)
print("even value at indexs: ", x1_search)

# Ví dụ tìm chỉ số index nơi có giá trị là lẻ (odd):
arr = np.array([1,2,6,4,6,3,6])
x2_search = np.where(arr%2 == 1)
print("odd value at indexs: ",x2_search)

Kết quả :

Even value at indexs: (array([1, 2, 3, 4, 6], dtype=int64),)
Odd value at indexs: (array([0, 5], dtype=int64),)

Tìm kiếm sử dụng hàm searchsorted()