Source code for featuretools.primitives.standard.transform.binary.and_primitive

import numpy as np
from woodwork.column_schema import ColumnSchema
from woodwork.logical_types import Boolean, BooleanNullable

from featuretools.primitives.base.transform_primitive_base import TransformPrimitive


class And(TransformPrimitive):
    """    Performs element-wise logical AND of two lists.

    Description:
        给定一个布尔值列表 X 和一个布尔值列表 Y,
        判断 X 中的每个值是否为 `True`,并且
        其对应的 Y 中的值是否也为 `True`.

    Examples:

        >>> _and = And()
        >>> _and([False, True, False], [True, True, False]).tolist()
        [False, True, False]"""

    name = "and"
    input_types = [
        [
            ColumnSchema(logical_type=BooleanNullable),
            ColumnSchema(logical_type=BooleanNullable),
        ],
        [ColumnSchema(logical_type=Boolean), ColumnSchema(logical_type=Boolean)],
        [
            ColumnSchema(logical_type=Boolean),
            ColumnSchema(logical_type=BooleanNullable),
        ],
        [
            ColumnSchema(logical_type=BooleanNullable),
            ColumnSchema(logical_type=Boolean),
        ],
    ]
    return_type = ColumnSchema(logical_type=BooleanNullable)
    commutative = True

    description_template = "whether {} and {} are true"

    def get_function(self):
        return np.logical_and

    def generate_name(self, base_feature_names):
        return "AND(%s, %s)" % (base_feature_names[0], base_feature_names[1])