标签: python arrays numpy vectorization
我有两个numpy数组分别表示'a'和'b'分别具有维度(327600,5)和(3,5)。我想对两个数组进行逐元素减法。如果没有循环,我该怎么做呢?
当我做[0] - b时,我得到一个(3,5)阵列。我想在减法后获得一个(327600,3,5)数组作为结果。
答案 0 :(得分:1)
a[:, np.newaxis] - b
您需要在形状中使用额外长度为1的轴来查看a,以便轴向右排列以进行广播。
a