如何在Python列表推导式中合理使用临时变量？

1. 海象运算符（Python 3.8+）：最直观的写法 #

Python 3.8引入的海象运算符:=完美解决了临时变量复用的问题，允许在表达式内部赋值并直接复用，无需重复调用split，也不用伪装循环变量：

nsms = [fields 
        for line in lines 
        if (fields := line.split(';')) 
        and len(fields) > 4 
        and fields[4] == 'NSM']

这里fields := line.split(';')会先执行拆分操作并将结果赋值给fields，后续的条件判断直接复用这个变量，逻辑清晰且高效。

2. 生成器表达式预处理（兼容Python 3.8以下） #

如果你的Python版本低于3.8，无法使用海象运算符，可以用生成器表达式先批量处理所有行的拆分结果，再进行过滤：

nsms = [fields 
        for fields in (line.split(';') for line in lines) 
        if len(fields) > 4 
        and fields[4] == 'NSM']

外层的生成器(line.split(';') for line in lines)会逐行执行拆分，每一行只调用一次split，之后直接对拆分后的fields进行条件判断，写法简洁且符合列表推导式的常规逻辑。

对比你当前的实现 #

你之前用for fields in [line.split(';')]的方式，本质是借助单元素列表的循环来创建临时变量，虽然能实现需求，但可读性不如上面两种方法——生成器表达式的写法更贴近常规推导逻辑，海象运算符则是专门为这类场景设计的语法糖。

内容的提问来源于stack exchange，提问作者Adrian McCarthy