Bagaimana untuk Menukar Tatasusunan Indeks kepada Tatasusunan Berekod Satu-Hot dalam NumPy?

Tatasusunan Indeks Pengekodan Satu Panas dalam NumPy

Dalam NumPy, menukar tatasusunan indeks 1D kepada tatasusunan 2D yang dikodkan satu-panas ialah tugas biasa. Sebagai contoh, memandangkan tatasusunan a dengan indeks [1, 0, 3], kami ingin mengekodnya sebagai:

b = [[0,1,0,0], [1,0,0,0], [0,0,0,1]]

Untuk mencapai ini, terdapat dua langkah utama:

1. Buat tatasusunan sifar: Cipta tatasusunan 2D b dengan lajur yang mencukupi (iaitu, a.max() 1) untuk menampung nilai dikodkan satu-panas. Tatasusunan hendaklah dimulakan dengan sifar.
2. Tetapkan nilai yang sesuai kepada 1: Untuk setiap baris i dalam b, tetapkan lajur a[i] kepada 1. Ini menunjukkan bahawa indeks asal a[i] hadir pada kedudukan i dalam tatasusunan berkod satu-panas.

Berikut ialah contoh kod untuk menggambarkan:

<code class="python">import numpy as np

a = np.array([1, 0, 3])
b = np.zeros((a.size, a.max() + 1))
b[np.arange(a.size), a] = 1

print(b)</code>

Output:

[[0. 1. 0. 0.]
 [1. 0. 0. 0.]
 [0. 0. 0. 1.]]

Kaedah ini secara berkesan menukar tatasusunan indeks kepada tatasusunan dikodkan satu panas, di mana setiap baris mewakili nilai dikodkan satu panas bagi indeks yang sepadan dalam a.

Atas ialah kandungan terperinci Bagaimana untuk Menukar Tatasusunan Indeks kepada Tatasusunan Berekod Satu-Hot dalam NumPy?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!