如何读取多个 zip 文件中的所有 csv 文件？

Question

我有一个包含许多 zip 文件的文件夹，这些 zip 文件中有多个 csv 文件。 有没有办法在 python 的一个数据框中获取所有 .csv 文件？ 或者我可以通过任何方式传递 zip 文件列表？

我目前正在尝试的代码是：

import glob
import zipfile
import pandas as pd

for zip_file in glob.glob(r"C:\Users\harsh\Desktop\Temp\data_00-01.zip"):
    # This is just one file. There are multiple zip files in the folder
    zf = zipfile.ZipFile(zip_file)
    dfs = [pd.read_csv(zf.open(f), header=None, sep=";", encoding='latin1') for f in zf.namelist()]
    df = pd.concat(dfs,ignore_index=True)
    print(df)

此代码适用于一个 zipfile，但我的文件夹中有大约 50 个 zip 文件，我想在一个数据帧中读取并连接这些 zip 文件中的所有 csv 文件。

谢谢

Answer 1

以下代码应满足您的要求（只需根据您的需要编辑dir_name）：

import glob
import zipfile
import pandas as pd

dfs = []
for filename in os.listdir(dir_name):
    if filename.endswith('.zip'):
        zip_file = os.path.join(dir_name, filename)
        zf = zipfile.ZipFile(zip_file)
        dfs += [pd.read_csv(zf.open(f), header=None, sep=";", encoding='latin1') for f in zf.namelist()]
df = pd.concat(dfs,ignore_index=True)