retain_variables_if_in_df

feature_engine.variable_handling.retain_variables_if_in_df(X, variables)

返回列表中存在于数据框中的变量子集。

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

参数

X: pandas 数据框，形状为 [n_samples, n_features]

数据集。

变量: 字符串, 整数或字符串或整数的列表。

要检查的变量名称。

返回

variables_in_df: 列表。

存在于 X 中的 变量 子集。

示例

>>> import pandas as pd
>>> from feature_engine.variable_handling import retain_variables_if_in_df
>>> 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")
>>> })
>>> vars_in_df = retain_variables_if_in_df(X, ['var_num', 'var_cat', 'var_other'])
>>> vars_in_df
['var_num', 'var_cat']