Python：将列表推导式转为Lambda函数以根据prop获取cat值

嘿，我来帮你搞定这个Lambda函数的转换问题！

首先先还原一下你的场景——假设你的原始数据列表大概是这样的：

my_item_list = [
    {'prop': 'ABC', 'cat': 'ABC Dir'},
    {'prop': 'DEF', 'cat': 'DEF Dir'},
    {'prop': 'GHI', 'cat': 'GHI Dir'}
]

你之前用列表推导式实现的逻辑大概是：

# 提取prop为'ABC'对应的cat值
result = [item['cat'] for item in my_item_list if item['prop'] == 'ABC'][0]

能拿到预期的ABC Dir，但想把这个逻辑塞进Lambda里却卡壳了对吧？

几种可行的Lambda实现方案 #

1. 推荐：高效处理单匹配+无匹配兜底

用next()配合生成器表达式，比列表推导式更高效（找到第一个匹配就停止遍历），还能设置无匹配时的默认值避免报错：

get_cat_by_prop = lambda target_prop, item_list: next(
    (item['cat'] for item in item_list if item['prop'] == target_prop),
    None  # 无匹配时返回None，你也可以改成自己需要的默认值
)

用法示例：

print(get_cat_by_prop('ABC', my_item_list))  # 输出: ABC Dir
print(get_cat_by_prop('XYZ', my_item_list))  # 输出: None

2. 简化版（仅适用于确定有匹配的场景）

如果你能100%保证目标prop一定存在于列表中，也可以直接把列表推导式塞进Lambda：

get_cat_by_prop = lambda target_prop, item_list: [item['cat'] for item in item_list if item['prop'] == target_prop][0]

但要注意：如果没有匹配项，这个写法会抛出IndexError，所以只适合边界情况明确的场景。

3. 返回所有匹配项（多结果场景）

如果你的列表可能有多个相同prop的项，想要返回所有对应的cat值，直接返回列表推导式结果即可：

get_all_cats_by_prop = lambda target_prop, item_list: [item['cat'] for item in item_list if item['prop'] == target_prop]

用法示例：

print(get_all_cats_by_prop('ABC', my_item_list))  # 输出: ['ABC Dir']

小提醒 #

Lambda虽然简洁，但如果后续需要扩展逻辑（比如增加匹配规则、数据校验），还是建议改用普通函数，可读性和可维护性会更好哦～

内容的提问来源于stack exchange，提问作者Me Okey