按索引更新列表元素

Question

我想使用list comprehension将5添加到我的特定列表索引中

输入

arr=[0,0,0,0,0]

产量

arr=[0,0,5,5,5]

我试过了

[arr[i]+=5 for i in range(2,4)]

但它给出了一个错误.

Answer 1

不要将列表推导用于副作用.list comp的目的是创建一个新列表.为此,我相信你可以在这里使用enumerate+range

l, u = 2, 4
arr = [x + 5 if i in range(l, u + 1) else x for i, x in enumerate(arr)]

print(arr)
[0, 0, 5, 5, 5]

在python3中,这应该是非常有效的,因为in对range对象的检查是O(1)时间.在python2上,执行布尔检查会更快(这是python3中的in检查range) -

arr = [x + 5 if l <= i <= u else x for i, x in enumerate(arr)]

但是,请记住,for循环是这里使用的最有效的方法.

for i in range(l, u + 1):
    arr[i] += 5

print(arr)
[0, 0, 5, 5, 5]

因为,