Na przykład, mam ndarray
czyli:Jak podzielić tablicę zgodnie z warunkiem w numpy?
a = np.array([1, 3, 5, 7, 2, 4, 6, 8])
Teraz chcę podzielić a
na dwie części, jedna jest wszystkie numery < 5, a druga jest wszystko> = 5:
[array([1,3,2,4]), array([5,7,6,8])]
Oczywiście mogę przechodzić a
i utworzyć dwie nowe tablice. Ale chcę wiedzieć, czy numpy oferuje lepsze sposoby?
Podobnie w przypadku tablic wielowymiarowych, np.
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[2, 4, 7]])
Chcę podzielić według pierwszej kolumnie < 3 i> = 3, którego wynikiem jest:
[array([[1, 2, 3],
[2, 4, 7]]),
array([[4, 5, 6],
[7, 8, 9]])]
Czy są jakieś lepsze sposoby zamiast biegu to? Dzięki.
Nevermind, ja goofed. To metoda, a nie nazwa globalna. Kontynuuj .. – Daenyth
Niesamowite krojenie! Nigdy nie znam tej metody. Czas dokładnie przeczytać numpy doc ... Dzięki! – Clippit
Czy 'arr [cond], arr [~ cond]' oznacza, że testuje każdy element tablicy dla tego samego warunku dwukrotnie? – endolith