Source code for featuretools.primitives.standard.aggregation.count_less_than

from woodwork.column_schema import ColumnSchema
from woodwork.logical_types import Integer

from featuretools.primitives.base.aggregation_primitive_base import AggregationPrimitive


[docs]class CountLessThan(AggregationPrimitive): """Determines the number of values less than a controllable threshold. Args: threshold (float): The threshold to use when counting the number of values less than. Defaults to 10. Examples: >>> count_less_than = CountLessThan(threshold=3.5) >>> count_less_than([1, 2, 3, 4, 5]) 3 """ name = "count_less_than" input_types = [ColumnSchema(semantic_tags={"numeric"})] return_type = ColumnSchema(logical_type=Integer, semantic_tags={"numeric"}) stack_on_self = False default_value = 0
[docs] def __init__(self, threshold=10): self.threshold = threshold
def get_function(self): def count_less_than(x): return x[x < self.threshold].count() return count_less_than