python - 在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}