Home/ Questions/Q 1143
Next
In Process
Calvin
  • 1
Explainer
Asked: 2023-07-01T07:59:14+00:00 In:

How can I add arrays in Numpy without using loops?

  • 1

A= array([1, 2, 3, 4, 2, 3, 0, 1, 5, 3, 1, 0])

In this array I want to add the values of first three arrays in list i.e., 1+2+3 = 6

Then I want to add the next three values i.e 4+2+3 = 9 and so on.

How to perform this array addition without loops?

numpyprogramspython
Share
Leave an answer

Leave an answer

Browse

1 Answer

  1. Pundit

    You can subtract the first sums from the second and subsequent sums by taking a linear subset of the cumulative sum:

    import numpy as np
A = np.array([1, 2, 3, 4, 2, 3, 0, 1, 5, 3, 1, 0])
R = A.cumsum()[2::3]
R[1:] -= R[:-1]
print(R)

    Output:

    [6 9 6 4]
    • 0