numpy.where() у Python
The numpy.where() функція повертає індекси елементів у вхідному масиві, де задана умова задовольняється.
Синтаксис: numpy.where(умова[, x, y])
Параметри:
хвороба : Якщо True, дає x, інакше y.
x, y: Цінності, з яких можна вибрати. x, y і умова повинні транслюватися в певну форму.Повернення:
вихід: [ndarray або кортеж ndarrays] Якщо вказано як x, так і y, вихідний масив містить елементи x, де умова має значення True, і елементи з y в інших місцях.Якщо вказано лише умову, поверніть кортеж condition.nonzero(), індекси, де умова має значення Істинна.
Код №1:
# Python program explaining> # where() function> > import> numpy as np> > np.where([[> True> ,> False> ], [> True> ,> True> ]],> > [[> 1> ,> 2> ], [> 3> ,> 4> ]], [[> 5> ,> 6> ], [> 7> ,> 8> ]])> |
Вихід:
array([[1, 6], [3, 4]])
Код №2:
# Python program explaining> # where() function> > import> numpy as np> > # a is an array of integers.> a> => np.array([[> 1> ,> 2> ,> 3> ], [> 4> ,> 5> ,> 6> ]])> > print> (a)> > print> (> 'Indices of elements <4'> )> > b> => np.where(a <> 4> )> print> (b)> > print> (> 'Elements which are <4'> )> print> (a[b])> |
Вихід:
[[1 2 3] [4 5 6]] Indices of elements <4 (array([0, 0, 0], dtype=int64), array([0, 1, 2], dtype=int64)) Elements which are <4 array([1, 2, 3])
