分别访问numpy数组中的元素

问题描述

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

解决方法

自从您标记了true：

numpy

输出：

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

,

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)