python - 从 numpy 数组中逐一减去行

Question

我有一个 2D numpy 数组 A。

我想从 A 中逐行减去每一行，并将逐行绝对和存储在数组中。

有没有一种方法可以不使用for循环来执行操作？下面是带有 for 循环的代码。

import numpy as np
A = np.random.randint(5,size=(8,9))
b = np.zeros(A.shape[1]);
for i in xrange(A.shape[0]):
    b = b + np.sum(np.absolute(A - A[i,:]), axis=0)

Answer 1

您可以使用broadcasting -

(np.abs(A[:,None,:] - A)).sum(axis=(0,1))

步骤:

(1) 保持最后一个轴对齐，得到 A 的两个版本:

Input1 (A[:,None,:])    :  M x 1 x N
Input2 (A)              :      M x N

获取这两个输入之间的绝对差，从而生成 3D 数组。

(2) 沿前两个轴求和以获得最终输出。