python - 单独访问 numpy 数组中的元素

Question

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

标签： pythonarraysnumpyelement

Answer 1

既然你标记了numpy：

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

输出：

array([ 5, 25, 61])