如何在 python 中打开 .zip 文件？

Question

为了打开 .zip 文件而不是 .gz，我必须在此代码中进行哪些更改？

import gzip
embeddingsIn = gzip.open(embeddingsPath, "r") if embeddingsPath.endswith('.gz') else open(embeddingsPath, encoding="utf8")

Answer 1

根据gzip documentation，没有简单或直接的方法可以做到这一点：

GzipFile 类读取和写入 gzip 格式的文件，[...] 请注意，该模块不支持 gzip 和 gunzip 程序可以解压缩的其他文件格式，例如 compress 和 pack 生成的文件格式。

一种可能的解决方法是使用zipfile module，它确实支持.zip 文件，并组合了一个自定义my_open 函数来打开.gz、.zip 和通用文件

import zipfile, gzip

def my_open(path):
    if zipfile.is_zipfile(path):
        return zipfile.ZipFile(path)
    elif path.endswith('.gz'):
        return gzip.GzipFile(path)
    else:
        return open(path, encoding="utf8")

作为说明，我使用了 ZipFile 和 GzipFile 构造函数，因为根据它们各自的文档（zipfile、gzipfile），它们在功能上都等同于 gzip.open，而 @ 中没有实现987654333@.