Bagaimana untuk Mengekstrak Elemen daripada Tatasusunan NumPy 2D Menggunakan Indeks daripada Tatasusunan Lain?

Mengekstrak Elemen daripada Tatasusunan 2D Menggunakan Indeks daripada Tatasusunan Lain

Dalam NumPy, kadangkala ia menjadi perlu untuk mengekstrak elemen tertentu daripada tatasusunan berbilang dimensi berdasarkan indeks yang disimpan dalam tatasusunan lain. Senario ini sering timbul apabila bekerja dengan struktur data seperti matriks jarang atau pilihan diindeks.

Masalah:

Pertimbangkan dua tatasusunan NumPy:

A = np.array([[0, 1], [2, 3], [4, 5]])
B = np.array([[1], [0], [1]])  # Index array

Matlamatnya adalah untuk mengekstrak satu elemen daripada setiap baris A, di mana elemen khusus ditentukan oleh indeks dalam baris B yang sepadan. Output yang dijangkakan hendaklah:

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

Penyelesaian:

1. Pengindeksan Tatasusunan Integer Tulen:

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

Kaedah ini melibatkan penggunaan keupayaan pengindeksan tatasusunan integer NumPy. Ia menjana julat indeks yang sepadan dengan baris A dan menggabungkannya dengan tatasusunan B yang diratakan untuk memilih elemen yang sesuai.

2. Transposing dan np.choose:

np.choose(B.ravel(), A.T)

Dalam pendekatan alternatif ini, anda transpose A untuk memadankan bentuk B dan kemudian gunakan np.choose untuk memilih elemen yang dikehendaki berdasarkan tatasusunan B yang diratakan.

3. Iterable Unpacking (Python >=3.6):

*A = A.T
C = np.array([*zip(*A)][i] for i in B.ravel())

Kaedah ini menggunakan iterable unpacking untuk menukar A kepada senarai baris dan kemudian melelaran ke atas baris A berdasarkan indeks dalam B kepada ekstrak elemen yang dikehendaki.

4. Senaraikan Pemahaman dan Penyiaran:

[A[i][j] for i, j in zip(range(A.shape[0]), B.ravel())]

Kefahaman senarai boleh digunakan untuk mencipta tatasusunan baharu dengan mengulangi elemen A dan B dan melaksanakan pemilihan mengikut unsur.

5. Pengindeksan Fancy (NumPy >=1.18):

A[np.stack([range(A.shape[0]), B.ravel()], axis=1)]

Pengindeksan mewah membolehkan operasi pengindeksan yang lebih cekap dan padat. Dalam kes ini, ia mencipta tatasusunan 2D dengan indeks baris dan indeks B, yang boleh digunakan untuk memilih elemen yang dikehendaki daripada A.

Penyelesaian yang paling sesuai bergantung pada keperluan khusus dan kekangan tugas, seperti kecekapan, kebolehbacaan dan keserasian dengan versi lama NumPy.

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