0

我有一个 n 个主键的列表：

primary_keys = ['pkey1', 'pkey2', ..., 'pkeyn']

我有一个键/值的字典：

kw = {'pkey1': 123, 'pkey2': 456, ..., 'pkeyn': 789, 'other': 'abc', 'values': 'def'}

我有一个实例列表：

instances = [obj, obj, obj]

kw我想在实例列表中找到其主键与我的字典中的主键值匹配的实例。

有没有一种简单的方法可以在 Python 中做到这一点？

注意：我很确定我不会有超过 2 个 PK，但我想允许更多，以防万一。

4

3 回答 3

1

如果只有一个匹配的实例，这样的事情应该可以工作。

next(inst for inst in instances
          if all(get_key_value(inst, pkey) == kw[pkey] for pkey in primary_keys))

Whereget_key_value是一个函数，它返回给定实例的给定键的值。如果键是实例的简单属性，则可以使用getattr，而如果实例是dict-like，则可以使用inst[pkey].

示例用法（实例简化为其主键集）：

>>> instances = [{'a': 5, 'b': 3}, {'a': 7, 'b': 8},  {'a': -1, 'b': 19}]
>>> primary_keys = ['a', 'b']
>>> kw = {'a': 7, 'b': 8}
>>> next(inst for inst in instances
...           if all(inst[pkey] == kw[pkey] for pkey in primary_keys))
{'a': 7, 'b': 8}

如果没有实例与键匹配，StopIteration则会引发 a，您可以轻松地用try块捕获。如果您想保留所有匹配的实例，您可以简单地将上面的内容转换为列表理解：

[inst for inst in instances
      if all(get_key_value(inst, pkey) == kw[pkey] for pkey in primary_keys)]
于 2013-03-27T08:48:12.197 回答
0

这样的事情怎么样？

match = []
for instance in instances:
    if instance.key in kw:
        match.append(instance)
于 2013-03-27T08:47:43.640 回答
0

这是一段带有测试的实际工作代码：

primary_keys = ['pkey1', 'pkey2', 'pkey3']

kw = {'pkey1': 123, 'pkey2': 456, 'pkey3': 789, 'other': 'abc', 'values': 'def'}

def obj_matches(obj):
    for key in kw.keys():
        if key.startswith('pkey') and getattr(obj, key) == kw[key]:
            return True
    return False

class Foo(object):    
    def __init__(self, pkey1, pkey2, pkey3):
        self.pkey1, self.pkey2, self.pkey3 = pkey1, pkey2, pkey3

instances = [Foo(123, 0, 0), Foo(0, 456, 0), Foo(0, 0, 789), Foo(1, 2, 3)]

matches = []
for obj in instances:
    if obj_matches(obj):
        matches.append(obj)


for m in matches:
    print m.pkey1, m.pkey2, m.pkey3
于 2013-03-27T08:58:01.867 回答