识别列表长度相似性的最佳方法

Question

我有一个字典，其键下包含列表：

dct = {'a': [1, 2, 3],
       'b': [1, 2, 3, 4],
       'c': [1, 2]}

识别列表长度是否相同的最佳方法是什么？

这是我的解决方案：

import itertools
len(set(itertools.imap(len, dct.viewvalues()))) == 1

True 如果相似，False 如果不是

UPD：参考@RaymondHettinger 建议将 map 替换为 itertools.imap

Answer 1

你的解决方案看起来不错。

如果您想稍微调整一下，请使用 itertools.imap() 而不是 map()。这会将内存占用缩减为 O(1)，而不是 O(n)。

Answer 2

首先，我会坚持使用 itervalues，它使用简单的评估。

其次，我会对依赖使用 set 持谨慎态度，因为它在遍历字典的每次迭代中都会查找集合中的值。超额的时间为 O(1)（在最坏的情况下为 O(n)，在我们的例子中，如果所有的都为 O(1)长度相同，如果所有长度不同，则为 O(n)）根据 文档。但很难评估使用 set 的开销。

在这种情况下我会使用all。当找到第一个 False 值时，all 失败。因此，长度的第一次不匹配将停止交互过程。而如果使用set，它会遍历所有列表直到最后，然后才将其长度与1进行比较。

>>> dct = {'a': [1, 2, 3],
       'b': [1, 2, 3, 4],
       'c': [1, 2]}
>>> lenght_1 = len(dct.itervalues().next())
>>> all(len(value)==lenght_1 for value in dct.itervalues())
False

>>> dct = {'a': [1, 2, 3],
       'b': [1, 2, 4],
       'c': [1, 2, 5]}
>>> lenght_1 = len(dct.itervalues().next())
>>> all(len(value)==lenght_1 for value in dct.itervalues())
True

可以通过使用相同的迭代器 it 来优化代码，该迭代器不会两次遍历第一个值：

>>> it = dct.itervalues()
>>> length_1 = len(next(it))
>>> all(len(value)==l1 for value in it)
True

Answer 3

注意：ovgolovin 的解决方案是好多了。我将这个答案留在这里，因为有讨论提到它。

您的解决方案很好，但您可以使用使用更少内存且更具可读性的生成器表达式：

len(set(len(x) for x in dct.viewvalues()))) == 1

Answer 4

正如 Michael J. Barber 在对 答案的评论中所建议的那样，这是使用 groupby 和 imap 来自 itertools 模块。

imap 只是将 len 函数应用于每个列表。

groupby 只是摸索相同长度的块中的值。

因此，如果有多个长度块，则长度是不同的。如果只有一个长度，则意味着列表的长度相同，第二次访问 groupby 迭代器应产生 StopIteration ，从而返回 None（next 函数的默认值）。

这段代码的最大好处是 imap 和 groupby 是用 C 编写的，而且速度相当快。

from itertools import imap,groupby

dct = {'a': [1, 2, 3],
       'b': [1, 2, 3, 4],
       'c': [1, 2]}

dct2 = {'a': [1, 2, 3],
       'b': [1, 2, 34],
       'c': [1, 2, 5]}

def check_lenghts(iterable):
    it = groupby(imap(len,iterable.itervalues()))
    next(it,None)
    return True if next(it,None)==None else False

print(check_lenghts(dct))
print(check_lenghts(dct2))