Rumah >pembangunan bahagian belakang >Tutorial Python >Bagaimana untuk Mencari Nilai Terdekat dalam Array NumPy?

Bagaimana untuk Mencari Nilai Terdekat dalam Array NumPy?

Barbara Streisand
Barbara Streisandasal
2024-11-29 21:00:13998semak imbas

How to Find the Closest Value in a NumPy Array?

Mencari Nilai Terdekat dalam Tatasusunan Numpy

Dalam numpy, mencari nilai terdekat kepada nilai sasaran yang diberikan boleh dicapai melalui proses yang mudah . Mari kita tentukan fungsi untuk memenuhi keperluan ini:

import numpy as np

def find_nearest(array, value):
    array = np.asarray(array)
    idx = (np.abs(array - value)).argmin()
    return array[idx]

Fungsi ini menerima tatasusunan dan nilai sasaran. Ia menukar tatasusunan kepada tatasusunan numpy dan mengira perbezaan mutlak antara setiap elemen tatasusunan dan nilai sasaran. Indeks elemen dengan perbezaan mutlak terkecil ditentukan menggunakan argmin(). Akhirnya, elemen pada indeks itu dikembalikan.

Contoh Penggunaan:

Mari kita jelaskan penggunaan fungsi find_nearest() dengan contoh:

array = np.random.random(10)
print(array)

print(find_nearest(array, value=0.5))

Dalam contoh ini, tatasusunan rawak 10 elemen dicipta. Fungsi find_nearest() kemudiannya digunakan untuk mencari elemen dalam tatasusunan yang paling hampir dengan 0.5. Output akan menjadi nilai terdekat dalam tatasusunan kepada 0.5.

Atas ialah kandungan terperinci Bagaimana untuk Mencari Nilai Terdekat dalam Array NumPy?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Kenyataan:
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn