[使用map()处理列表时,我想在lambda内访问项目的索引。我该怎么办?
例如
ranked_users = ['jon','bob','jane','alice','chris']
user_details = map(lambda x: {'name':x, 'rank':?}, ranked_users)
在以上示例中如何获得每个用户的排名?
使用enumerate:
In [3]: user_details = [{'name':x, 'rank':i} for i,x in enumerate(ranked_users)]
In [4]: user_details
Out[4]:
[{'name': 'jon', 'rank': 0},
{'name': 'bob', 'rank': 1},
{'name': 'jane', 'rank': 2},
{'name': 'alice', 'rank': 3},
{'name': 'chris', 'rank': 4}]
PS。我的第一个答案是
user_details = map(lambda (i,x): {'name':x, 'rank':i}, enumerate(ranked_users))
我强烈建议尽可能在map
和lambda
上使用列表推导或生成器表达式。列表理解更具可读性,并且往往可以更快地启动。
或者,您可以使用list comprehension而不是map()和lambda。
ranked_users = ['jon','bob','jane','alice','chris']
user_details = [{'name' : x, 'rank' : ranked_users.index(x)} for x in ranked_users]
输出:
[{'name': 'jon', 'rank': 0}, {'name': 'bob', 'rank': 1}, {'name': 'jane', 'rank': 2}, {'name': 'alice', 'rank': 3}, {'name': 'chris', 'rank': 4}]
列表推导非常强大,并且比faster和map
的组合还lambda
。
我认为问题是关于映射函数的,并且由于将元组参数放在lambda lambda (i,x)
上而导致的语法错误,因此首选答案在一定程度上是正确的>
枚举的想法很好,正确的解决方案是:
map(lambda x: {'name':x[1], 'rank':x[0]}, enumerate(ranked_users))
以及一些将速度与理解力进行比较的时间:
def with_map(): ranked_users = range(10 ** 6) list(map(lambda x: {'name': x[1], 'rank': x[0]}, enumerate(ranked_users))) def by_comprehension(): ranked_users = range(10 ** 6) [{'name': x, 'rank': i} for i, x in enumerate(ranked_users)] from timeit import timeit time_with_map = timeit(with_map, number=10) time_with_comprehension = timeit(by_comprehension, number=10) print('list comprehension is about %.2f x faster than map in this test case' % (time_with_map/time_with_comprehension))
测试结果:在此测试案例中,列表理解比map快约1.31倍
实际上,与在地图中使用枚举元组相比,这是一个更为优雅,冗长的解决方案(因为元组索引)。 Map可以将更多的Iterables作为参数,因此让我们使用它。