numpy.ndarray.compress

метод

ndarray.compress(condition, axis=None, out=None)

Возвращает выбранные срезы этого массива вдоль заданной оси.

Для получения полной документации см. numpy.compress.

См. также

numpy.compress

эквивалентная функция