2D ndarray：行操作

Question

我有2D numpy的數組，例如形狀：

>>> a.shape 
(48, 160) 

和我想要做的元件或每行之間簡單的操作，象a[0] - a[1]但針對所有其它行中的每一行。

我知道如何簡單地通過使用for循環和迭代行做到這一點，但我想知道如果有一些numpy的切片特定指令，可以做到這一點，而無需使用for循環

Answer 1

您可以使用廣播魔法做這個。

import numpy as np 
a = np.arange(12).reshape((4, 3)) 
b = np.arange(15).reshape((5, 3)) 
diff = a[np.newaxis, :, :] - b[:, np.newaxis, :] 
diff.shape 
# (5, 4, 3) 

This是一個很好的廣播教程。在這種情況下，我做了（1，4，3）和b（5,1,3），我得到的結果是（5,4,3），a和b中每一行的差異。