我有一个列表a = [1, 2, 3, 4, 5]
,现在我想为1
之后的每个元素添加index 2
,即a[2] + 1
,a[3] + 1
,{ {1}}。
我最终想要a[4] + 1
。
这是什么最恐怖的方式?
答案 0 :(得分:8)
>>> 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])
答案 1 :(得分:1)
for i in range(2, len(a)) :
a[i] += 1