Python 3.x：列表元素的条件乘法实现问题

我需要实现对列表中的元素进行条件乘法：当元素i≥3时乘以3，当元素i<3时乘以2。尝试了以下函数，但未得到预期输出[2,4,9,12,15,18]，请帮忙解决问题。

用户尝试的代码：

def special_multiply(list, multiplicator,multiplicator_2):
    for i in list:
        if i >= 3:
            result = [i* multiplicator for i in list]
        if i < 3:
            result = [i * multiplicator_2 for i in list]
        return result

print("Special multiply:", special_multiply([1,2,3,4,5,6],2,3))

问题原因 #

1. 提前返回导致循环终止：函数内的return result写在循环内部，第一次循环拿到第一个元素（值为1）后就直接返回结果，后续元素根本没有被处理。
2. 列表推导式覆盖逻辑错误：每次判断都会生成一个覆盖全列表的推导结果，比如第一个元素满足i<3时，会把整个列表所有元素都乘以multiplicator_2，完全忽略了条件判断的针对性。

修复后的代码 #

def special_multiply(input_list, multiplier_low, multiplier_high):
    return [num * multiplier_high if num >= 3 else num * multiplier_low for num in input_list]

print("Special multiply:", special_multiply([1,2,3,4,5,6], 2, 3))

关键优化点 #

• 用三元表达式结合列表推导式，一次性对每个元素单独做条件判断和乘法操作，逻辑简洁且高效。
• 将参数名list改为input_list，避免与Python内置的list类型重名，规避潜在语法冲突。
• 移除多余的循环和提前返回逻辑，确保列表中所有元素都能被正确处理。

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