我有两个不同大小的 Numpy 数组 A (n x 1) 和 B (m x 1)。我想从 A 的所有元素中减去 B 的每个元素。因此,结果矩阵 C (m x n) 的元素应计算为 c(i,j) = A(i)-B(j)
。有没有使用 Numpy 的直接无循环计算?
最佳答案
广播:
A = np.array([1,2,3,4,5])
B = np.array([5,4,2,7])
A - B[:, np.newaxis]
输出:
array([[-4, -3, -2, -1, 0],
[-3, -2, -1, 0, 1],
[-1, 0, 1, 2, 3],
[-6, -5, -4, -3, -2]])
关于python - 如何从另一个数组中减去一个数组的每个元素?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57114574/