← retour aux snippets

sklearn: make_column_selector

Sélecteurs de colonnes par dtype/noms.

objectif

Sélecteurs de colonnes par dtype/noms.

code minimal

from sklearn.compose import make_column_selector as selector
import numpy as np, pandas as pd
df = pd.DataFrame({"a":["x","y"], "b":[1.0, 2.0]})
sel = selector(dtype_include=object)
cols = sel(df)
print(cols == ["a"])

utilisation

from sklearn.compose import make_column_selector as selector
import pandas as pd
df = pd.DataFrame({"a":["x"], "b":[1.0]})
print(selector(pattern="^b$")(df) == ["b"])

variante(s) utile(s)

from sklearn.compose import make_column_selector as selector
print(callable(selector()))

notes

  • Combinez avec ColumnTransformer pour pipelines robustes.