python – 有效地减去不同形状的numpy数组

python – 有效地减去不同形状的numpy数组

使用numpy的优秀广播规则,您可以从形状(5,3)数组X中减去形状(3,)数组v X – v 结果是形状(5,3)数组,其中每行i是差X [i] -v. 有没有办法从X中减去一个形状(n,3)数组w,以便从整个数组X中减去w的每一行而不显式使用循环? 您需要使用 None/np.newaxis扩展X的尺寸