| 3302 | |
| 3303 | @derived_from(pd.DataFrame) |
| 3304 | def drop(self, labels=None, axis=0, columns=None, errors="raise"): |
| 3305 | if columns is None and labels is None: |
| 3306 | raise TypeError("must either specify 'columns' or 'labels'") |
| 3307 | |
| 3308 | axis = _validate_axis(axis) |
| 3309 | |
| 3310 | if axis == 1: |
| 3311 | columns = labels or columns |
| 3312 | elif axis == 0 and columns is None: |
| 3313 | raise NotImplementedError( |
| 3314 | "Drop currently only works for axis=1 or when columns is not None" |
| 3315 | ) |
| 3316 | return new_collection(expr.Drop(self, columns=columns, errors=errors)) |
| 3317 | |
| 3318 | def to_parquet(self, path, **kwargs): |
| 3319 | from dask.dataframe.dask_expr.io.parquet import to_parquet |