两个numpy数组的元素减法

时间:2015-11-19 00:26:53

标签: python arrays numpy vectorization

我有两个numpy数组分别表示'a'和'b'分别具有维度(327600,5)和(3,5)。我想对两个数组进行逐元素减法。如果没有循环,我该怎么做呢?

当我做[0] - b时,我得到一个(3,5)阵列。我想在减法后获得一个(327600,3,5)数组作为结果。

1 个答案:

答案 0 :(得分:1)

a[:, np.newaxis] - b

您需要在形状中使用额外长度为1的轴来查看a,以便轴向右排列以进行广播。