Selecting certain indices from numpy array and adding 2

Question

import numpy as np
arr = np.array([1,2,3,4,5,6,7,8,9])

How to add 2 to arr[0:2] and arr[5:6] so the final result is:

arr[3,4,3,4,5,8,7,8,9]

Onyambu · Accepted Answer · 2022-05-20 18:58:23Z

2

Note that you could create an array that has all the indices to the values that need to be modified:

arr[np.r_[0:2, 5:6]] += 2
print(arr)

Out:

array([3, 4, 3, 4, 5, 8, 7, 8, 9])

answered May 20, 2022 at 18:57

Onyambu

80.3k3 gold badges29 silver badges65 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

jotjern · Accepted Answer · 2022-05-20 18:53:59Z

1

Very straight forward :)

import numpy as np

arr = np.array([1,2,3,4,5,6,7,8,9])

arr[0:2] += 2
arr[5:6] += 2

print(arr)

answered May 20, 2022 at 18:53

jotjern

4927 silver badges19 bronze badges