python - 在 defaultdict(list) python 中查找元素的数量

Question

一个简单的问题？我想知道是否有一种 pythonic 方式来获取以下内容。

我有一个默认字典(列表)

 foo = {"a":[1,2,3], "b":[3]....} and so on.

是否有一种 pythonic 方式来编写一个函数，它只返回列表长度超过“n”的键

例如

 def return_keys_based_on_value_length(num,dictionary):
    key_list = []
    for k,v in dictionary:
         if len(v)>= num:
            key_list.append(k)
    return key_list

有没有pythonic的方法来做到这一点？谢谢

Answer 1

foo = {"a":[1,2,3], "b":[3], "c":[1,2], "d":[1,2,3,4]}

n = 2

my_list = [key for key,val in foo.iteritems() if len(val) > n]

结果:

>>> my_list
['a', 'd']