Bagaimana untuk Mengekstrak Elemen daripada Tatasusunan NumPy Menggunakan Pengindeksan Tatasusunan Integer?

Pengekstrakan Berindeks dengan Tatasusunan NumPy

Apabila bekerja dengan tatasusunan berbilang dimensi dalam NumPy, selalunya perlu mengekstrak elemen tertentu berdasarkan indeks. Dalam hal ini, senario tertentu mungkin timbul apabila tatasusunan (B) berfungsi sebagai indeks untuk dimensi kedua tatasusunan lain (A).

Pertimbangkan contoh dengan dua tatasusunan A dan B:

A = np.array([[0,1], 
              [2,3], 
              [4,5]])
B = np.array([[1],
              [0],
              [1]], dtype='int')

Matlamat kami adalah untuk mengekstrak satu elemen daripada setiap baris A, di mana indeks elemen ditentukan oleh elemen yang sepadan dalam B. Output yang dikehendaki ialah:

C = np.array([[1],
              [2],
              [5]])

Penyelesaian Menggunakan Integer Pengindeksan Tatasusunan:

Daripada menyiarkan B, kita boleh menggunakan pengindeksan tatasusunan integer semata-mata NumPy:

A[np.arange(A.shape[0]),B.ravel()]

Di sini, np.arange(A.shape[0]) mencipta tatasusunan indeks yang sepadan dengan bilangan baris dalam A, manakala B.ravel() meratakan B menjadi tatasusunan 1D. Dengan menggabungkan ini, kita boleh mengekstrak elemen yang sesuai daripada A.

Sample Run:

print(A)
print(B)
print(A[np.arange(A.shape[0]),B.ravel()])

Output:

[[0 1]
 [2 3]
 [4 5]]
[[1]
 [0]
 [1]]
[1 2 5]

Pendekatan ini menyediakan penyelesaian yang ringkas dan cekap kepada masalah pengekstrakan kami.

Atas ialah kandungan terperinci Bagaimana untuk Mengekstrak Elemen daripada Tatasusunan NumPy Menggunakan Pengindeksan Tatasusunan Integer?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!