Jak mogę usunąć niektóre określone elementy z tablicy numpy? Powiedzmy, że mam taki kod:
import numpy as np
a = np.array([1,2,3,4,5,6,7,8,9])
Następnie chcę usunąć 3,4,7 z a. Wiem tylko jaki jest indeks wartości (index = [2,3,6]).
0
1 odpowiedź
Użyj numpy.delete () - zwraca nową tablicę z usuniętymi pod-tablicami wzdłuż usuniętej osi
numpy.delete(a, index)
W przypadku Twojego konkretnego pytania będzie to wyglądać w ten sposób:
import numpy as np
a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
index = [2, 3, 6]
new_a = np.delete(a, index)
print(new_a) #Printuje `[1, 2, 5, 6, 8, 9]`
Zauważ, że numpy.delete () zwraca nową tablicę, ponieważ skalary tablic są niezmienne, podobne do łańcuchów w Pythonie, więc za każdym razem, gdy wprowadzana jest do niej zmiana, tworzony jest nowy obiekt. Aby zacytować dokumenty delete ():
0