在Python中向列表的一部分添加偏移量？

Question

我有一个列表a = [1, 2, 3, 4, 5]，现在我希望向从index 2开始的每个元素中添加一个1，即a[2] + 1、a[3] + 1、a[4] + 1。

那就是我最后想要a = [1, 2, 4, 5, 6]。

什么是最毕达通的方式？

Answer 1

>>> a = [1, 2, 3, 4, 5]
>>> a[2:] = [x+1 for x in a[2:]]
>>> a
[1, 2, 4, 5, 6]

对于numpy数组：

>>> a = np.array([1,2,3,4,5])
>>> a[2:] += 1
>>> a
array([1, 2, 4, 5, 6])

Answer 2

for i in range(2, len(a)) :
    a[i] += 1