如何优化Python列表推导式？转换函数遇重复False与无序问题求助

核心问题点 #

• 逻辑偏离原函数：原函数仅当目标词完全不在列表中时才返回False，而你的列表推导式会给每个不匹配的位置生成False，再用set()去重，这完全违背了原函数的判断逻辑。
• 无序问题根源：set()是无序集合，存入其中的元素会丢失原本的顺序，所以返回的位置会乱序。
• 返回值类型不一致：原函数返回的是有序列表或False，你的版本返回的是集合，类型不符合预期。

正确的列表推导式实现 #

要贴合原函数逻辑，应该先判断目标词是否存在，再收集匹配位置：

def song_lyrics(a_list, lyric):
    if lyric not in a_list:
        return False
    return [pos for pos, word in enumerate(a_list) if word == lyric]

the_wall = 'we dont need no education we dont need no thought control no we dont'.split()
result = song_lyrics(the_wall, 'we')
print(result)  # 输出: [0, 5, 12]

代码说明 #

1. 先做整体判断：如果lyric不在a_list里，直接返回False，和原函数逻辑一致。
2. 用列表推导式仅收集匹配词的位置，不需要生成False，自然也不需要用set()去重，保留了位置的原始顺序。
3. 返回值类型和原函数一致：存在匹配时返回有序列表，不存在时返回False。

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