numpy.ndarray.compress

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

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

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

См. также