python - 使用字典 python 从不同路径导入 csv

Question

我正在尝试从不同的路径导入 CSV，以字典指示指定的路径。

import csv
import os
import pandas as pd

PATH = Path("../../Toxic-comment-classification")
PATH.mkdir(exist_ok=True)

args = {
    "data_dir": PATH,
}

class MultiLabelTextProcessor(DataProcessor):

    def __init__(self, data_dir):
        self.data_dir = data_dir
        self.labels = None


    def get_train_examples(self, data_dir, size=-1):
        filename = 'train.csv'
        data_df = pd.read_csv(os.path.join(data_dir, filename))
        return(data_df)

a = MultiLabelTextProcessor(args.values())
print(a.get_train_examples(a.data_dir))

但是，当我尝试运行时出现以下错误:

TypeError: expected str, bytes or os.PathLike object, not dict_values

我理解 args.values() 给了我一个 dict_object。我怎样才能将我的字典中的值作为字符串？

Answer 1

显然您正在使用Python 3。args.values()不会给你一个 dict 对象，而是一个 dict_values ，Python 2 中的等效项是列表。但是，在您的情况下， list/dict_values 只有一项，即 args.values() == [PATH] (有点)

我打赌你期待 args是一个只有一个条目的字典，所以你想提取 PATH内容。你可以简单地做list(args.values())[0]在上面。或者，在我看来更好的方法是使用 args['data_dir']

os.path.join(data_dir, filename)与 data_dir列表或 dict_values 类型和 filename字符串类型导致您的代码出现问题。