如何用Python列表推导式实现开放读码框终止密码子计数功能？

需求说明 #

实现nextStop2函数，规则如下：

• 每次从序列头部取3个字符作为密码子
• 遇到属于终止密码子的三字符组合时立即停止计算
• 返回终止密码子之前的所有字符总长度，终止密码子及后续序列不计入
• 示例：执行nextStop2('AAAAAAAGTGGGTGCTAGGTTGGC')应返回15

原代码错误原因 #

你最初的写法无法运行，是因为存在两个语法/逻辑问题：

• Python的生成器表达式、列表推导式本身没有内置的中途终止语法，你写的end_of_loop()不是合法语法，无法实现普通for循环中break的效果
• 直接用GeneticCodeStop[Seq[i:i+3]]取值时，如果密码子不在终止密码子字典的键中，会直接抛出KeyError，而非你预期的返回非'X'的值

优化实现方案 #

方案1：基于itertools.takewhile的简洁实现（推荐） #

该写法符合Python语法规范，性能和可读性都较好，逻辑和你提供的正确for循环完全一致：

from itertools import takewhile
def nextStop2(Seq):
    # 用集合存储终止密码子，判断存在的效率更高
    GeneticCodeStop = {'TAA', 'TAG', 'TGA'}
    # takewhile遇到第一个不满足条件的密码子时自动停止迭代
    valid_codons = takewhile(lambda codon: codon not in GeneticCodeStop, 
                            (Seq[i:i+3] for i in range(0, len(Seq), 3)))
    return len(''.join(valid_codons))

方案2：纯推导式实现（无需额外导包） #

通过直接查找第一个终止密码子的起始位置实现，写法更精简：

def nextStop2(Seq):
    GeneticCodeStop = {'TAA', 'TAG', 'TGA'}
    # 找到第一个终止密码子的起始索引，没找到则取序列总长度
    stop_index = next((i for i in range(0, len(Seq), 3) if Seq[i:i+3] in GeneticCodeStop), len(Seq))
    return stop_index

你提供的普通for循环实现是逻辑最直观的写法，性能和上述方案没有明显差异，如果没有必须用推导式的强制要求，保留原有for循环写法也是很好的选择。

内容的提问来源于stack exchange，提问作者xin hwa