Access elements in a numpy array individually

Question

Say I have a numpy array [[1,2],[3,4],[5,6]], how do I do element-wise mathematics such that I could iterate over each XY pair to get X^2 + Y^2 for each pair?

Quang Hoang · Accepted Answer · 2020-09-21 17:37:10Z

2

Since you tagged numpy:

(np.array(a)**2).sum(-1)

Output:

array([ 5, 25, 61])

answered Sep 21, 2020 at 17:37

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

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

Comments

Sharan Ravigopal · Accepted Answer · 2020-09-21 17:44:38Z

0

import numpy as np

arr1 = np.array([[1,2],[3,4],[5,6]])
    
rows = arr1.shape[0] 
cols = arr1.shape[1]
    
ans = []
   
for x in range(rows):
    answer = arr1[x,0]**2 + arr1[x,1]**2
    ans.append(answer)
print(ans)

answered Sep 21, 2020 at 17:44

Sharan Ravigopal

1

1 Comment

hpaulj Over a year ago

Why did you iterate? Since you have an array just do: (arr1**2).sum(axis=1)

Collectives™ on Stack Overflow

Access elements in a numpy array individually

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related