Skip to content

Commit 664a1cb

Browse files
authored
[Docs] Add description of select_columns to guide (#6119)
Add `select_columns` to guide
1 parent cb8c5de commit 664a1cb

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

docs/source/process.mdx

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,28 @@ Dataset({
195195
})
196196
```
197197

198+
Conversely, [`~Dataset.select_columns`] selects one or more columns to keep and removes the rest. This function takes either one or a list of column names:
199+
200+
```py
201+
>>> dataset
202+
Dataset({
203+
features: ['sentence1', 'sentence2', 'label', 'idx'],
204+
num_rows: 3668
205+
})
206+
>>> dataset = dataset.select_columns(['sentence1', 'sentence2', 'idx'])
207+
>>> dataset
208+
Dataset({
209+
features: ['sentence1', 'sentence2', 'idx'],
210+
num_rows: 3668
211+
})
212+
>>> dataset = dataset.select_columns('idx')
213+
>>> dataset
214+
Dataset({
215+
features: ['idx'],
216+
num_rows: 3668
217+
})
218+
```
219+
198220
### Cast
199221

200222
The [`~Dataset.cast`] function transforms the feature type of one or more columns. This function accepts your new [`Features`] as its argument. The example below demonstrates how to change the [`ClassLabel`] and [`Value`] features:

0 commit comments

Comments
 (0)