mindspore.dataset.Dataset.get_col_names

mindspore.dataset.Dataset.get_col_names()[源代码]

返回数据集对象中包含的列名。

返回:

list,数据集中所有列名组成列表。

样例:

>>> import mindspore.dataset as ds
>>> dataset = ds.GeneratorDataset([i for i in range(10)], "column1")
>>> col_names = dataset.get_col_names()
>>> print(col_names)
['column1']