python - Pandas 用部分通配符读取csv文件

Question

我正在尝试编写一个导入文件的脚本，然后对该文件执行某些操作并将结果输出到另一个文件。

df = pd.read_csv('somefile2018.csv')

上面的代码工作得很好。但是，我想避免在代码中对文件名进行硬编码。

脚本将在包含 script.py 的文件夹(目录)中运行和几个 csv 文件。

我试过以下方法:

somefile_path = glob.glob('somefile*.csv')

df = pd.read_csv(somefile_path)

但是我得到以下错误:

ValueError: Invalid file path or buffer object type: <class 'list'>

Answer 1

glob 返回一个列表，而不是一个字符串。 read_csv 函数将字符串作为输入来查找文件。试试这个:

for f in glob('somefile*.csv'):
    df = pd.read_csv(f)
    ...
    # the rest of your script