Python：在numpy數組上按元素明智的操作構建矩陣

Question

給定兩個numpy數組（arr1和arr2），我想構建一個矩陣，用於存儲arr1中每個元素與arr2中每個元素的差。 即：

my_matrix = [arr1-i for i in arr2]

但是，隨着陣列變大，這種情況開始變慢。 我試圖像這樣利用numpy的良好性能：

arr1_mtx = np.array([arr1]*len(arr2))
arr2_mtx = np.array([arr1]*len(arr2)).T
my_matrix = arr1_mtx-arr2_mtx

我對Python還是很陌生，所以我不確定這是否是構建該矩陣的最pythonic和最有效的方法。 有小費嗎？

提前致謝！

Answer 1

您可以將其中一個數組整形為2d數組，然后使用numpy 廣播 ：

arr1[:,None] - arr2

---

arr1 = np.array([1,2])
arr2 = np.array([3,4,5])
​
arr1[:,None] - arr2
#array([[-2, -3, -4],
#       [-1, -2, -3]])