如何查看Hugging Face datasets.Dataset内容？求类似Pandas df.head()的方法

方法1：切片获取前n条数据 #

和Python列表切片逻辑一致，直接用dataset[:n]即可获取前n条记录，返回字典格式，键对应列名，值为该列的前n个数据：

print(dataset[:1])  # 获取前1条数据

输出示例：

{'a': [1], 'b': [1]}

方法2：转回Pandas DataFrame后使用head() #

既然数据集是从Pandas DataFrame转换而来，也可以再转回去，直接用熟悉的head()方法：

dataset.to_pandas().head()

方法3：按索引访问单条/批量数据 #

通过索引直接访问单条记录，返回单条数据的字典：

print(dataset[0])  # 获取第1条数据，输出为{'a': 1, 'b': 1}

如果要批量遍历，用take()方法指定数量：

for item in dataset.take(2):  # 遍历前2条数据
    print(item)

方法4：查看底层Arrow数据 #

对于小型数据集，可直接查看底层的Arrow表格数据，展示完整内容：

print(dataset.data)

内容的提问来源于stack exchange，提问作者ClaudiaR