Python中如何根据列表长度为if语句设置多分支判断条件

你当前硬编码多分支的写法维护成本极高，每新增一种短语长度就要追加判断分支，还存在重复调用方法、重复写计数逻辑、长短语匹配漏判的问题，完全可以用统一逻辑覆盖所有长度场景，不需要逐个写if/elif分支。

核心逻辑本质是通用的：短语按空格拆分后，除第一个词外的所有剩余词，需要和parser.words数组中从index+1位置开始的等长序列完全匹配，匹配成功就给对应path的计数加1。

优化后代码（无额外依赖，兼容任意短语长度） #

for index in indices:
    phrase_words = phrase.split(" ")
    # 取出需要匹配的后缀部分（首词之外的所有词）
    match_suffix = phrase_words[1:]
    suffix_len = len(match_suffix)
    # 用列表切片直接对比，不需要逐个写and条件判断
    if parser.words[index+1 : index+1+suffix_len] == match_suffix:
        # 简化计数逻辑，无需额外判断key是否存在
        rang[path] = rang.get(path, 0) + 1

额外优化可选 #

如果可以引入标准库，初始化rang的时候可以用defaultdict进一步简化计数逻辑：

from collections import defaultdict
rang = defaultdict(int)

# 循环内匹配成功后直接写即可，不需要get判断
rang[path] += 1

优化收益 #

• 完全适配任意长度的短语，后续不需要为新的长度追加任何判断分支
• 消除所有重复代码，总代码量缩减70%以上
• 修复原代码的隐藏bug：原else分支在短语长度大于4时，只会校验前3个后缀词，漏检后续位置的词会导致错误计数，优化后不存在该问题
• 避免重复调用phrase.split(" ")，减少无意义的性能损耗

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