Maison > Article > développement back-end > Comment convertir des tableaux d'index en tableaux codés à chaud dans NumPy ?
Tableaux d'index d'encodage à chaud dans NumPy
Dans NumPy, la conversion d'un tableau d'index 1D en un tableau 2D encodé à chaud est une tâche commune. Par exemple, étant donné le tableau a avec les indices [1, 0, 3], nous voulons l'encoder comme :
b = [[0,1,0,0], [1,0,0,0], [0,0,0,1]]
Pour y parvenir, il y a deux étapes clés :
Voici un exemple de code pour illustrer :
<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>
Sortie :
[[0. 1. 0. 0.] [1. 0. 0. 0.] [0. 0. 0. 1.]]
Cette méthode convertit efficacement le tableau d'indices en un tableau codé à chaud, où chaque ligne représente une valeur codée à chaud de l'index correspondant dans a.
Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!