我可以可视化 datasets.Dataset 的内容吗？答案

【问题标题】：Can I visualize the content of a datasets.Dataset?我可以可视化 datasets.Dataset 的内容吗？
【发布时间】：2022-07-27 00:02:43
【问题描述】：

我正在使用 Huggingface datasets 库从 pandas 数据框中加载数据集。代码是这样的：

from datasets import Dataset
import pandas as pd
df = pd.DataFrame({"a": [1], "b":[1]})
dataset = Dataset.from_pandas(df)

一切都很顺利，但是，我想仔细检查加载的Dataset 的内容。我一直在寻找类似于我们在 Pandas 中的df.head() 的东西，但我在官方Huggingface documentation 上一无所获。有没有办法“读取”甚至部分加载数据集的内容？

做一个简单的print(dataset) 不显示内容，只显示一些高级信息：

Dataset({
    features: ['a', 'b'],
    num_rows: 1
})

【问题讨论】：

【解决方案1】：

答案比你想象的要简单。做吧

print(dataset[i])

其中i 是行号（第一个是0）。

输出将是一个字典，其中特征作为键，行的内容作为值。

print(dataset[0])

<<< {
"a": [1],
"b": [1]
}

【讨论】：