单独访问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)