模拟 python 中 findWhere() 的行为

Question

Underscore 有一个方便的小函数findWhere()，可用于在列表中查找特定结构，例如

myList = [
  {'name': 'Thor'},
  {'name': 'Odin'},
  {'name': 'Freya'},
  {'name': 'Skadi'}
];
findWhere(myList, {'name': 'Skadi'});

结果：[{'name': 'Skadi'}]

更好的例子：

my_list = [
   {'name': 'Thor',
    'occupation': 'God of Thunder',
    'favorite color': 'MY HAMMER'}
   {'name': 'Skadi',
   'occupation': 'Queen of the Ice Giants',
   'favorite color': 'purpz'}
  ]
findWhere(my_list, {'name': 'Skadi'})

结果：

[{'name': 'Skadi',
'occupation': 'Queen of the Ice Giants',
'favorite color': 'purpz'}]

唉，我在 python 中找不到类似的东西。实现相同功能的 Python 方式是什么？

Answer 1

您可以简单地将其定义为生成器：

def find_where(iterable, dct):
    for item in iterable:
        if all(item[key] == value for key, value in dct.items()):
            yield item

my_list = [
  {'name': 'Thor', 'age': 23},
  {'name': 'Odin', 'age': 42},
  {'name': 'Freya', 'age': 50},
  {'name': 'Skadi', 'age': 23},
]

print list(find_where(my_list, {'age': 23}))

输出：

[{'age': 23, 'name': 'Thor'}, {'age': 23, 'name': 'Skadi'}]

另请参阅all()并列出推导式以了解有关表达式“内容”的详细信息。