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

Question

我有一个 dict 在其键下包含列表：

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

注意：ovgolovin's solution 更好。我把这个答案留在这里是因为有讨论提到它。

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

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

Answer 3

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

其次，我会警惕依赖使用set，因为它会在每次遍历字典时查找集合中的值。根据docs，它是超额的O(1)（如果所有长度都相同，则O(n) 在我们的情况下是O(1)，如果所有长度不同，则为O(n)）。但是很难评估使用 set 的开销。

在这种情况下，我会使用all。 all 在找到第一个 False 值时失败。因此，长度的第一个不匹配将停止交互过程。而如果使用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 4

正如 Michael J. Barber 在 cmets 中向 the answer 建议的那样，这里是使用 itertools 模块中的 groupby 和 imap 的代码。

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

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

因此，如果有多个长度块，则长度是不同的。如果只有一个长度的chuck，这意味着列表的长度相同，第二次访问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))