如何将一个numpy数组的每个子numpy数组与两个numpy数组相乘和除

Question

我试图将一个 numpy 数组的每个子 numpy 数组与两个 numpy 数组相乘和除。

我有一个形状为[100, 5] 的numpy 数组x，两个形状为(5,) 的numpy 数组y 和y。

我正在尝试更改张量的值： 对于每个子numpy数组w以及x中的axis = 0，它应该具有[1, 5]的形状，我想做(w - y)*z。

我的想法是对x 进行for循环并选择其中的每个子数组来执行此操作，然后重建原始数组。但是，这可能效率不高。

Answer 1

这应该可行。

(x - y) * z

样本：

>>> x.shape,y.shape, z.shape
((10L, 5L), (5L,), (5L,))
>>> x
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24],
       [25, 26, 27, 28, 29],
       [30, 31, 32, 33, 34],
       [35, 36, 37, 38, 39],
       [40, 41, 42, 43, 44],
       [45, 46, 47, 48, 49]])
>>> y
array([0, 1, 2, 3, 4])
>>> z
array([1, 2, 3, 4, 5])

>>> (x-y)*z
array([[  0,   0,   0,   0,   0],
       [  5,  10,  15,  20,  25],
       [ 10,  20,  30,  40,  50],
       [ 15,  30,  45,  60,  75],
       [ 20,  40,  60,  80, 100],
       [ 25,  50,  75, 100, 125],
       [ 30,  60,  90, 120, 150],
       [ 35,  70, 105, 140, 175],
       [ 40,  80, 120, 160, 200],
       [ 45,  90, 135, 180, 225]])