Misalkan Anda memiliki ndarray z, dan ingin mendapatkan indeks berikut

z_idxes1 = np.argwhere(z == i)
z_idxes2 = np.argwhere(z != i)

Bisakah kita melakukan ini dalam satu panggilan ke beberapa fungsi numpy daripada dua panggilan argwhere?

1
Ralph B. 6 Mei 2018, 16:01

1 menjawab

Jawaban Terbaik

Tidak dalam satu panggilan, tetapi Anda dapat menggunakan operator logika bukan untuk mendapatkan negasi dari array topeng Anda.

mask = (z == i)
z_idxes1 = np.argwhere(mask)

z_idxes2 = np.argwhere(~mask)
1
Kasravnd 6 Mei 2018, 14:52