python - 我如何知道 .npz 文件是否已压缩？

Question

给定一个来自 np.savez 或 np.savez_compressed 的 .npz 文件，当它被 np.load 加载时，是否有任何如何检查文件是否被压缩？

我试着看看 docs和 GitHub 。它没有告诉我任何内容，只是告诉我文件是如何压缩的。

Answer 1

np.load 返回一个 NpzFile 对象。

• 它有一个 zip attribute这是 ZipFile对象。

• 其中有 infolist()返回 ZipInfo 列表的方法对象。

• 其中有一个 compress_type 属性，其中包含所使用的压缩方法。

如果使用np.savez，则压缩类型为ZIP_STORED ，如果使用np.savez_compressed，则压缩类型为ZIP_DEFLATED (relevant source code)。

总结一下:

import numpy
import zipfile

def is_compressed(npz_file):
    zip_infos = npz_file.zip.infolist()
    if len(zip_infos) == 0:
        raise RuntimeError("Did not find ZipInfos unexpectedly")
    compress_type = zip_infos[0].compress_type
    if compress_type == zipfile.ZIP_STORED:
        return False
    elif compress_type == zipfile.ZIP_DEFLATED:
        return True
    else:
        raise ValueError("Unexpected compression type")

# Example
a = numpy.array([1, 2, 3])
numpy.savez("uncompressed.npz", a)
numpy.savez_compressed("compressed.npz", a)
u = numpy.load("uncompressed.npz")
c = numpy.load("compressed.npz")
print(is_compressed(u))  # False
print(is_compressed(c))  # True