Python遍历列表满足条件后填充剩余元素报错解决及列表推导式实现

报错原因与原代码修改 #

你遇到的TypeError: can only assign an iterable错误，是因为Python对列表切片赋值时，右侧必须传入可迭代对象，直接填写整数1不符合语法要求。
同时原逻辑还有优化空间：触发填充条件后无需继续遍历剩余元素，可以直接终止循环。
修改后的代码如下：

array = [0, 0, 0, 0, 0, 0, 0, 0]
for i in range(len(array)):
    if i % 2 == 0:
        # 右侧生成和切片长度一致的全1列表
        array[i:] = [1] * (len(array) - i)
        # 填充完成直接终止循环，避免无效遍历
        break
print(array)

运行后即可得到预期输出[1, 1, 1, 1, 1, 1, 1, 1]。

列表推导式实现方案 #

可以用列表推导式实现，思路是先定位到第一个满足触发条件的索引位置，再根据索引判断元素值：

array = [0, 0, 0, 0, 0, 0, 0, 0]
# 先找到第一个满足条件的索引，无符合条件的情况则默认不触发填充
trigger_idx = next((i for i in range(len(array)) if i % 2 == 0), len(array))
# 列表推导式生成结果
array = [1 if idx >= trigger_idx else 0 for idx in range(len(array))]
print(array)

如果你的触发条件是针对元素值而非索引，只需要把i % 2 == 0替换成对应的元素判断逻辑即可。

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