如何用兩個numpy數組將numpy數組的每個子numpy數組相乘和除

Question

我試圖將一個numpy數組的每個子numpy數組與兩個numpy數組相乘和除。

我有一個形狀為[100, 5] 100，5 [100, 5]的numpy數組x ，兩個都為形狀(5,) numpy數組y和y 。

我試圖更改張量的值：對於x中軸= 0的每個子numpy數組w ，它的形狀應為[1, 5] ，我想做(w - y)*z 。

我的想法是在x循環並選擇其中的每個子數組來執行此操作，然后重建原始數組。 但是，這可能沒有效率。

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]])