如何将Pandas DataFrame转换为Huggingface Dataset对象？

要把Pandas DataFrame转换成Hugging Face的Dataset，直接使用datasets库提供的方法即可，具体操作如下：

1. 安装依赖库 #

如果还未安装datasets，先执行安装命令：

pip install datasets

2. 转换代码示例 #

import pandas as pd
from datasets import Dataset

# 定义你的DataFrame
df = pd.DataFrame({"foo": ["bar", "baz"]})

# 执行转换
dataset = Dataset.from_pandas(df)

# 查看转换结果
print(dataset)

执行后会输出类似内容：

Dataset({
    features: ['foo'],
    num_rows: 2
})

如果需要保留DataFrame的索引，可添加keep_index=True参数：

dataset = Dataset.from_pandas(df, keep_index=True)

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