Python根据实例变量从列表中获取实例

Question

给定一个实例列表，比如clients，我试图根据单个实例变量screenName 的值从列表中提取一个项目。我知道我可以做到：

for client in clients:
  if client.screenName = search:
    return client

但是没有循环有没有更好的方法呢？

感谢您的帮助:)

Answer 1

你可以使用filter

try:
    filter(lambda client: client.screenName == search, clients)[0]
except IndexError:
    # handle error. May be use a default value

Answer 2

我会使用list comprehensions。假设这是你的 Client 类：

>>> class Client:
...    def __init__(self, screenName):
...        self.screenName = screenName

如果我得到这个客户列表：

>>> l = [Client('a'), Client('b'), Client('c')]

...我可以得到一个列表，其中只包含给定名称的客户：

>>> [e for e in l if e.screenName == 'b']
[<__main__.Client instance at 0x2e52b0>]

现在，只需获取第一个 - 并且假定只有 - 元素：

>>> [e for e in l if e.screenName == 'b'][0]
<__main__.Client instance at 0x2e52b0>
>>> c = [e for e in l if e.screenName == 'b'][0]
>>> c.screenName
'b'

这很短，恕我直言，很优雅，但效率可能会降低，因为列表推导将遍历所有列表。如果您确实想避免这种开销，您可以使用括号而不是方括号来获取生成器而不是新列表：

>>> g = (e for e in l if e.screenName == 'b')
>>> g
<generator object <genexpr> at 0x2e5440>
>>> g.next()
<__main__.Client instance at 0x2e52b0>

但是，请注意next() 方法只能调用一次。

HTH！

Answer 3

您可以使用generator expression，

client=next(client for client in clients if client.screenName == search)

但不是你还在循环，只是以不同的方式。

注意：如果没有客户端满足条件client.screenName == search，那么上面将引发StopIteration 异常。这与您的 for-loop 不同，后者退出循环而不返回任何内容。

根据您的情况，引发异常可能比静默失败要好。

如果您不想使用默认值而不是 StopIteration 异常，则可以使用 next 的 2 参数版本：

client=next(client for client in clients if client.screenName == search, 
            default_value)

Answer 4

为此使用字典：

假设：

d[screeName] = client

你可以这样做：

return d[search]  

Answer 5

如果clients 是dict，那么您可以使用clients[search]。如果列表中元素的顺序很重要，那么您可以使用来自collections 的OrderedDict。

Answer 6

关于这个话题的最佳讨论是link

return find(lambda client: client.screenName == search, clients)

这需要您定义一个通用的查找函数，该函数适用于所有类型的列表，如下所示：

def find(f, seq):
  """Return first item in sequence where f(item) == True."""
  for item in seq:
    if f(item): 
      return item