如何使用 lambda 获取列表中过滤项目的索引？

Question

我有一份水果清单[{'name': 'apple', 'qty': 233}, {'name': 'orange', 'qty': '441'}]

当我使用 lambda list(filter(lambda x: x['name']=='orange', fruits)) 过滤 orange 的列表时，我得到了正确的字典，但我无法获取字典的索引。索引应该是 1 而不是 0。

如何获得过滤项目的正确索引？

Answer 1

您可以使用列表推导和 enumerate() 代替：

>>> fruits = [{'name': 'apple', 'qty': 233}, {'name': 'orange', 'qty': '441'}]
>>> [(idx, fruit) for idx, fruit in enumerate(fruits) if fruit['name'] == 'orange']
[(1, {'name': 'orange', 'qty': '441'})]

就像在 cmets 中发布的 @ChrisRands 一样，您也可以通过为您的 fruits 列表创建一个枚举对象来使用 filter：

>>> list(filter(lambda fruit: fruit[1]['name'] == 'orange', enumerate(fruits)))
[(1, {'name': 'orange', 'qty': '441'})]
>>> 

以下是这两种方法的一些时间安排：

>>> setup = \
      "fruits = [{'name': 'apple', 'qty': 233}, {'name': 'orange', 'qty': '441'}]"
>>> listcomp = \
     "[(idx, fruit) for idx, fruit in enumerate(fruits) if fruit['name'] == 'orange']"
>>> filter_lambda = \
     "list(filter(lambda fruit: fruit[1]['name'] == 'orange', enumerate(fruits)))"
>>> 
>>> timeit(setup=setup, stmt=listcomp)
1.0297133629997006
>>> timeit(setup=setup, stmt=filter_lambda)
1.6447856079998928
>>>