mindspore.dataset.Dataset.project

mindspore.dataset.Dataset.project(columns)[源代码]

从数据集对象中选择需要的列，并按给定的列名的顺序进行排序。未指定的数据列将被丢弃。

参数：

columns (Union[str, list[str]]) - 要选择的数据列的列名列表。

返回：

Dataset，project操作后的数据集对象。

样例：

>>> # dataset is an instance object of Dataset
>>> columns_to_project = ["column3", "column1", "column2"]
>>>
>>> # Create a dataset that consists of column3, column1, column2
>>> # in that order, regardless of the original order of columns.
>>> dataset = dataset.project(columns=columns_to_project)