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

Question

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

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

dong so 的最Pythonic 方式是什么？

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])