Wie bekomme ich jedes Element von numpy array?

Question

Ich habe eine Reihe wie folgt:

Keys, die einige Werte gespeichert werden. zum Beispiel Schlüssel [2,3,4,7,8]

Wie bekomme ich den Index von 4 und speichere den Index in einer int-Variablen?

Zum Beispiel ist der Indexwert von 4 2, also wird 2 in einer int-Variablen gespeichert.

Ich habe versucht, mit folgenden Codesegment

//enter code here 

for i in np.nditer(Keys): 
     print(keys[i]); 

//enter code here 

Ich verwende Python 3.5 Spyder 3.5.2 Anaconda 4.2.0

Answer 1

Es gibt mehrere Möglichkeiten. Wenn die Liste nicht zu groß ist, dann:

where_is_4 = [e for i,e in enumerate(Keys) if i==4][0] 

Was dies tut, ist es über die Liste mit Aufzählungs Schleifen und erstellt eine Liste, die den Wert des Enumerators enthält jedes Mal, wenn der Wert ‚4‘ auftritt.

Answer 2

Warum nicht einfach tun:

for i in range(len(Key)): 
    if (Key[ i ] == 4): 
     print(i) 

Answer 3

Sie alle Indizes finden können, wo der Wert 4 mit:

>>> keys = np.array([2,3,4,7,8]) 
>>> np.flatnonzero(keys == 4) 
array([2]) 

Answer 4

Ist keys eine Liste oder numpy Array

keys = [[2,3,4,7,8] # or 
keys = np.array([2,3,4,7,8]) 

Sie Sie müssen nicht iterieren, um die Elemente von beiden zu sehen. Aber Sie können tun

for i in keys: 
    print(i) 
for i in range(len(keys)): 
    print(keys[i]) 
[i for i in keys] 

diese für beide arbeiten.

Wenn Sie den Index des Wertes 4 wollen, muss die Liste eine Methode:

keys.index(4) 

für das Array

np.where(keys==4) 

ist ein nützliches Stück Code. Auch

np.in1d(keys, 4) 
np.where(np.in1d(keys, 4)) 

Vergessen Sie über np.nditer. Das ist für fortgeschrittene Programmierung, nicht Routine-Iteration.

Answer 5

Es gibt eine native Methode namens where.

Es wird ein Array der Indizes zurückgegeben, wo eine bestimmte Bedingung wahr ist. So können Sie nur den ersten Eintrag wählen, wenn die Liste nicht leer ist:

N = 4 
indicies = np.where(x==N)[0] 
index = None 
if indicies: 
    index = indicies[0] 

Answer 6

Verwendung von numpy.where(condition) wird hier eine gute Wahl sein. Aus dem folgenden Code können Sie die Position von 4 erhalten.

import numpy as np 
keys = np.array([2,3,4,7,8]) 
result = np.where(keys==4) 
result[0][0]