python - 查找多个 .npz 文件的最大形状

Question

我有许多形状可能有所不同的 .npz 文件，我想找到哪个文件的形状最大。 npz 中有 2 个数组，我正在寻找第二个数组中最大的一个。以下代码片段有效，但返回形状所需的时间比我预期的要长。这是实现这一目标的最有效方法吗？我担心缩放，因为目前需要几秒钟才能找到最大形状[1]，而且我只循环 4 个数组

frameMax =0 
for f in npzs:
    d = np.load(f,mmap_mode='r')
    if d['arr_0'].shape[1]>frameMax:
        frameMax = d['arr_0'].shape[1]
    d=None

Answer 1

请记住，I/O 操作可能相对较慢。也就是说，您可以使用内置 max 将查找最大值的逻辑减少到以下内容:它将在 O(n) 中运行时间，并消除了您所做的作业的需要:

frameMax = max([np.load(f,mmap_mode='r')['arr_0'].shape[1] for f in npzs])