查找分类变量和数值变量

feature_engine.variable_handling.find_categorical_and_numerical_variables(X, variables=None)

在数据框或列表中查找数值变量和分类变量。

该函数返回两个列表；第一个列表包含对象类型或分类类型的变量名称，第二个列表包含数值变量的名称。

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

参数

Xpandas 数据框，形状为 = [n_samples, n_features]

数据集

变量list, 默认=None

如果 None，该函数将找到 X 中的所有分类变量和数值变量。或者，它将找到 X 中的分类变量和数值变量，从给定的列表中选择。

返回

变量: 元组

包含分类变量的列表和数值变量的列表的元组。

示例

>>> import pandas as pd
>>> from feature_engine.variable_handling import (
>>>   find_categorical_and_numerical_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_cat, var_num = find_categorical_and_numerical_variables(X)
>>> var_cat, var_num
(['var_cat'], ['var_num'])

返回类型

Tuple[List[Union[str, int]], List[Union[str, int]]]