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

Question

我有许多形状可能不同的 .npz 文件，我想找出哪个文件具有较大的形状。 npzs 中有 2 个数组，我正在寻找第二个数组中最大的一个。以下 sn-p 有效，但返回形状所需的时间比我预期的要长。这是实现这一目标的最有效方法吗？我担心缩放，因为目前需要几秒钟才能找到最大形状[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])