skrub.Expr.skb.drop

Expr.skb.drop(cols)

Drop some columns.

cols can be a column name or a list of column names, but also a skrub selector. Importantly, the exact list of columns that match the selector is stored during fit and then this same list of columns is dropped during transform.

Parameters:

colsstr, list of strings, or skrub selector

The columns to select

Returns:

dataframe without the dropped columns

Examples

>>> import skrub
>>> from skrub import selectors as s
>>> X = skrub.X(skrub.toy_orders().X)
>>> X
<Var 'X'>
Result:
―――――――
   ID product  quantity        date
0   1     pen         2  2020-04-03
1   2     cup         3  2020-04-04
2   3     cup         5  2020-04-04
3   4   spoon         1  2020-04-05
>>> X.skb.drop(['ID', 'date'])
<Apply DropCols>
Result:
―――――――
  product  quantity
0     pen         2
1     cup         3
2     cup         5
3   spoon         1
>>> X.skb.drop(s.string())
<Apply DropCols>
Result:
―――――――
   ID  quantity
0   1         2
1   2         3
2   3         5
3   4         1