numpy.extract

numpy.извлечь(условие, arr)

Возвращает элементы массива, удовлетворяющие некоторому условию.

Это эквивалентно np.compress(ravel(condition), ravel(arr)). Если условие является логическим np.extract эквивалентно arr[condition].

Обратите внимание, что place делает прямо противоположное extract.

Параметры:

условиеarray_like

Массив, ненулевые или True элементы которого указывают на элементы arr для извлечения.

arrarray_like

Входной массив того же размера, что и условие.

Возвращает:

извлечьndarray

Массив ранга 1 значений из arr где условие равно True.

Смотрите также

take, put, copyto, compress, place

Примеры

Попробуйте в вашем браузере!

>>> import numpy as np
>>> arr = np.arange(12).reshape((3, 4))
>>> arr
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> condition = np.mod(arr, 3)==0
>>> condition
array([[ True, False, False,  True],
       [False, False,  True, False],
       [False,  True, False, False]])
>>> np.extract(condition, arr)
array([0, 3, 6, 9])

Если условие является булевым:

>>> arr[condition]
array([0, 3, 6, 9])

НазадОткрыть во вкладке