查找分类变量

feature_engine.variable_handling.find_categorical_variables(X)

返回一个包含数据框中所有分类变量名称的列表。请注意，被转换为对象类型且可以解析为日期时间的变量将被排除。

更多详情请参阅 用户指南。

参数

Xpandas 数据框，形状为 = [样本数, 特征数]

数据集

返回

变量: 列表

分类变量的名称。

示例

>>> import pandas as pd
>>> from feature_engine.variable_handling import find_categorical_variables
>>> X = pd.DataFrame({
>>>     "var_num": [1, 2, 3],
>>>     "var_cat": ["A", "B", "C"],
>>>     "var_date": pd.date_range("2020-02-24", periods=3, freq="T")
>>> })
>>> var_ = find_categorical_variables(X)
>>> var_
['var_cat']

返回类型

List[Union[str, int]]