Python按多条件修改数组元素的实现问题及报错解决

问题背景 #

需要创建新数组，对原数组元素按以下规则修改：

• 元素小于3时保留原值
• 3≤元素<6时减1
• 元素≥6时减2

原尝试代码及报错：

array = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
for i in array:
    if i < 3:
        arrayn = array
    elif i < 6:
        arrayn = array - 1
    elif i < 10:
        arrayn = array - 2
print(arrayn)

报错信息：

File "C:\test.py", line 6, in <module>
    arrayn = array - 1
             ~~~~~~^~~
TypeError: unsupported operand type(s) for -: 'list' and 'int'

期望输出：[0, 1, 2, 2, 3, 4, 4, 5, 6, 7, 8]

错误原因 #

1. 直接对整个列表array执行减法操作，Python不支持列表与整数的算术运算，这是触发TypeError的核心原因。
2. 循环逻辑错误：每次循环都直接将整个原列表赋值给arrayn，没有逐个处理元素并构建新数组。

修正方案 #

方案1：使用循环逐个处理元素 #

初始化空列表，遍历原数组每个元素，按条件处理后添加到新列表：

array = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
arrayn = []
for i in array:
    if i < 3:
        arrayn.append(i)
    elif 3 <= i < 6:
        arrayn.append(i - 1)
    else:  # 元素≥6的情况
        arrayn.append(i - 2)
print(arrayn)

方案2：使用列表推导式（更简洁） #

利用Python的列表推导式一行完成逻辑，更符合Pythonic风格：

array = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
arrayn = [i if i < 3 else i-1 if 3 <= i <6 else i-2 for i in array]
print(arrayn)

两种方案运行后都会输出期望结果：[0, 1, 2, 2, 3, 4, 4, 5, 6, 7, 8]

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