Maison > Article > développement back-end > Comment convertir un tableau d'index en un tableau codé à chaud dans NumPy ?
Conversion d'un tableau d'indices en un tableau codé One-Hot dans NumPy
Souvent, il devient nécessaire de transformer un tableau 1D d'indices dans un tableau 2D où chaque ligne représente un encodage à chaud de l'index correspondant dans le tableau d'origine.
Exemple :
Ayons un tableau 1D d'indices ' a' :
<code class="python">a = np.array([1, 0, 3])</code>
Nous visons à créer un tableau 2D 'b' où chaque ligne est un encodage à chaud de l'index correspondant dans 'a' :
<code class="python">b = np.array([[0, 1, 0, 0], [1, 0, 0, 0], [0, 0, 0, 1]])</code>
Solution :
Pour réaliser cette transformation, nous pouvons utiliser les étapes suivantes :
<code class="python">b = np.zeros((a.size, a.max() + 1))</code>
<code class="python">b[np.arange(a.size), a] = 1</code>
Sortie :
<code class="python">[[ 0. 1. 0. 0.] [ 1. 0. 0. 0.] [ 0. 0. 0. 1.]]</code>L'exécution de ce code produit le tableau codé à chaud souhaité 'b' :
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!