You need to enable JavaScript to run this app.
最新活动
大模型
产品
解决方案
定价
生态与合作
支持与服务
开发者
了解我们

如何使用map与lambda函数提取嵌套列表中所有子列表的第i个元素?

解决嵌套列表提取子列表第i个元素的map+lambda问题

我来帮你搞定这个问题~先拆解下你原代码的问题,再给你几个可行的正确实现方式:

原代码的问题根源

你定义的extract = lambda x,i :x[i]是一个需要两个参数的函数,但map函数只会依次把a中的每个子列表作为单个参数传给第一个函数参数。而你写的extract(i = 1)会直接执行这个lambda(此时缺少必填的x参数,自然会报错),而不是传递一个能接收子列表的函数给map

正确的实现方式

这里有几种灵活的写法,都能得到你想要的[2,5,7]结果:

方式1:用闭包包装固定参数

用一个新的lambda帮你传递固定的i=1参数给原extract函数,让map可以正常调用:

extract = lambda x,i :x[i]
a = [[1,2,3],[4,5,6],[6,7,8]]
b = list(map(lambda x: extract(x, 1), a))
print(b)  # 输出: [2,5,7]

方式2:给lambda设置默认参数

调整lambda的定义,把i设为带默认值的参数,这样map只需要传入子列表x即可:

a = [[1,2,3],[4,5,6],[6,7,8]]
b = list(map(lambda x, i=1: x[i], a))
print(b)  # 输出: [2,5,7]

方式3:最简洁的写法(直接指定索引)

如果i是固定值,完全可以省略单独的extract函数,直接在lambda中写索引逻辑:

a = [[1,2,3],[4,5,6],[6,7,8]]
b = list(map(lambda x: x[1], a))
print(b)  # 输出: [2,5,7]

这些写法都能让map正确遍历每个子列表,提取出你需要的对应索引元素~

内容的提问来源于stack exchange,提问作者Deepak Gurumoorthy

火山引擎 最新活动