Bagaimana untuk Memilih Elemen daripada Tatasusunan Numpy Memuaskan Pelbagai Keadaan menggunakan Fungsi `where`?

Numpy Where Function dan Multiple Condition

Soalan ini berkaitan dengan Numpy where fungsi dan pemilihan elemen berdasarkan berbilang syarat.

Isu dan Penerangan Masalah

Pengguna ingin menggunakan dua syarat (lebih besar daripada dan kurang daripada) untuk memilih elemen daripada tatasusunan (dists) yang berada dalam julat tertentu. Walau bagaimanapun, menggunakan fungsi where sebagai (np.where(dists >= r)) dan (np.where(dists <= r dr)) menghasilkan hasil yang tidak dijangka.

Penyelesaian

Kaedah 1: Menggabungkan Syarat menjadi Kriteria Tunggal

Dalam kes khusus ini, adalah disyorkan untuk menggabungkan dua syarat menjadi satu kriteria:

dists[abs(dists - r - dr/2.) <= dr/2.]

Pendekatan mudah ini menyemak sama ada nilai mutlak perbezaan antara dist dan pusat julat (r dr/2) adalah kurang daripada atau sama dengan separuh lebar julat (dr/2).

Kaedah 2: Menggunakan Pengindeksan Fancy

Sebagai alternatif, seseorang boleh menggunakan pengindeksan mewah untuk memilih elemen terus daripada tatasusunan asal menggunakan topeng boolean:

dists[(dists >= r) & (dists <= r + dr)]

Faedah ini kaedah ialah ia menggunakan pengendali logik dari segi unsur (& dan |) untuk menggabungkan syarat, menghasilkan topeng binari yang mengenal pasti elemen yang memenuhi kriteria.

Penjelasan Isu

Pendekatan asal menggunakan (np.where(dists >= r)) dan (np.where(dists <= r dr)) gagal kerana np.where mengembalikan senarai indeks, bukan tatasusunan boolean. Pendaratan antara dua senarai nombor menghasilkan penilaian senarai kedua, bukan perbandingan logik.

Atas ialah kandungan terperinci Bagaimana untuk Memilih Elemen daripada Tatasusunan Numpy Memuaskan Pelbagai Keadaan menggunakan Fungsi `where`?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!