在 PYTHON 中将多个文件读入单独的数据帧

Question

我想知道在 python 中是否有一种方法可以从一个文件夹中读取多个 CSV 文件并通过文件名分配给单独的数据框。下面的代码会抛出一个错误，但要显示我粘贴它的点

import glob
for filename in glob.glob('*.csv'):
    index = filename.find(".csv")
    if "test" in filename:
        filename[:index]) = pd.read_csv(filename)

Answer 1

我相信您需要通过文件名创建dictionary of DataFrame 和keys：

d = {}
for filename in glob.glob('*.csv'):
    if "test" in filename:
        d[filename[:-4]] = pd.read_csv(filename)

什么是相同的：

d = {f[:-4]: pd.read_csv(f) for f in glob.glob('*.csv') if "test" in f}

如果只需要文件名，可以使用：

d = {os.path.basename(f).split('.')[0]:pd.read_csv(f) for f in glob.glob('*.csv') if "test" in f}