【发布时间】:2015-11-19 00:26:53
【问题描述】:
我有两个 numpy 数组,分别是尺寸为 (327600,5) 和 (3,5) 的“a”和“b”。我想对两个数组进行逐元素减法。我将如何在没有循环的情况下执行此操作?
当我执行 a[0] - b 时,我得到一个 (3,5) 数组。我想得到一个 (327600,3,5) 数组作为减法后的结果。
【问题讨论】:
标签: python arrays numpy vectorization
我有两个 numpy 数组,分别是尺寸为 (327600,5) 和 (3,5) 的“a”和“b”。我想对两个数组进行逐元素减法。我将如何在没有循环的情况下执行此操作?
当我执行 a[0] - b 时,我得到一个 (3,5) 数组。我想得到一个 (327600,3,5) 数组作为减法后的结果。
【问题讨论】:
标签: python arrays numpy vectorization
a[:, np.newaxis] - b
您需要查看 a 的形状,并在形状中添加一个额外的长度为 1 的轴,以便轴对齐以进行广播。
【讨论】: