Heim  >  Artikel  >  Backend-Entwicklung  >  Wie konvertiere ich ein Array von Indizes in ein One-Hot-codiertes Array in NumPy?

Wie konvertiere ich ein Array von Indizes in ein One-Hot-codiertes Array in NumPy?

Barbara Streisand
Barbara StreisandOriginal
2024-11-01 00:09:28521Durchsuche

How to Convert an Array of Indices to a One-Hot Encoded Array in NumPy?

Konvertieren eines Arrays von Indizes in ein One-Hot-codiertes Array in NumPy

Oft ist es notwendig, ein 1D-Array von Indizes zu transformieren in ein 2D-Array, wobei jede Zeile eine One-Hot-Codierung des entsprechenden Index im ursprünglichen Array darstellt.

Beispiel:

Lassen Sie uns ein 1D-Array von Indizes haben ' a':

<code class="python">a = np.array([1, 0, 3])</code>

Unser Ziel ist es, ein 2D-Array „b“ zu erstellen, wobei jede Zeile eine One-Hot-Codierung des entsprechenden Index in „a“ ist:

<code class="python">b = np.array([[0, 1, 0, 0], [1, 0, 0, 0], [0, 0, 0, 1]])</code>

Lösung:

Um diese Transformation zu erreichen, können wir die folgenden Schritte verwenden:

  1. Null-Array-Erstellung:
    Erstellen Sie ein Null-Array Array 'b' mit genügend Spalten, um den maximalen Indexwert in 'a' aufzunehmen, plus eins ('a.max() 1').
<code class="python">b = np.zeros((a.size, a.max() + 1))</code>
  1. Eins -Hot-Codierung:
    Setzen Sie für jede Zeile „i“ im Array die „a[i]“-te Spalte auf 1. Dieser Schritt wandelt jeden Index in „a“ in eine One-Hot-codierte Zeile in um 'b'.
<code class="python">b[np.arange(a.size), a] = 1</code>

Ausgabe:

Die Ausführung dieses Codes erzeugt das gewünschte One-Hot-codierte Array 'b':

<code class="python">[[ 0.  1.  0.  0.]
 [ 1.  0.  0.  0.]
 [ 0.  0.  0.  1.]]</code>

Das obige ist der detaillierte Inhalt vonWie konvertiere ich ein Array von Indizes in ein One-Hot-codiertes Array in NumPy?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!

Stellungnahme:
Der Inhalt dieses Artikels wird freiwillig von Internetnutzern beigesteuert und das Urheberrecht liegt beim ursprünglichen Autor. Diese Website übernimmt keine entsprechende rechtliche Verantwortung. Wenn Sie Inhalte finden, bei denen der Verdacht eines Plagiats oder einer Rechtsverletzung besteht, wenden Sie sich bitte an admin@php.cn