Numpy, neues Array basierend auf einem Array innerhalb eines Arrays

Question

Ich bin ein Looping über ein Array, um Daten zu extrahieren, aber es ist zu langsam. Gibt es eine Funktion in Numpy, die das schneller macht? Ich muss ein neues Array basierend auf Werten aus einem Array innerhalb eines Arrays erstellen.

Erstellen Sie zum Beispiel ein Array mit Autos aus USA.

input: array = [['ford', 'USA'], ['volkswagen', 'Germany'], ['jeep', 'USA']] 

    output: new_array = [['ford', 'USA'], ['jeep', 'USA']] 

Answer 1

ein Array von String dtype Unter der Annahme, können Sie die zweite Spalte und vergleichen Sie mit dem String 'USA' ein boolean-Array zu erhalten schneiden aus. Dieses boolesche Array könnte dann zur Indizierung in das Array verwendet werden, indem boolean-indexing verwendet wird, um gültige Zeilen auszuwählen und uns die gewünschte Ausgabe zu geben.

Somit würde die Implementierung einfach sein -

array[array[:,1] == 'USA']