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

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

Reading multiple files into separate data frames in PYTHON

python

glob

pandas