乘以numpy矩阵并得到一些结果

Question

我想用这个矩阵操作：

(a b)
(c d)
(e f)

并得到：

(a*a+b*b c*c+d*d e*e+f*f)

或任何矩阵，例如：

(a b c)
(d e f)
(g h i)
(j k l)

并获得

(a*a+b*b+c*c d*d+e*e+f*f g*g+h*h+i*i j*j+k*k+l*l)

如何使它普遍适用于任何矩阵？

Answer 1

IIUC，你想要平方和吗？

只需执行：

(a**2).sum(1)

示例：

import numpy as np

a = np.array([[ 0,  1,  2],
              [ 3,  4,  5],
              [ 6,  7,  8],
              [ 9, 10, 11]])

(a**2).sum(1)

输出：

array([  5,  50, 149, 302])

Multiply numpy matrices and get some of the results