单独访问 numpy 数组中的元素

Question

假设我有一个 numpy 数组 [[1,2],[3,4],[5,6]]，我该如何进行逐元素数学运算，以便我可以遍历每个 XY 对以获得每个对的 X^2 + Y^2？

Answer 1

自从您标记 numpy：

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

输出：

array([ 5, 25, 61])

Answer 2

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)