numpy數組結構與廣播

Question

我想將具有不同但廣播兼容陣列的陣列堆疊在一起。 給定7x5,7x1,1x5和1x1陣列，我想做類似的事情

 a475 = mkarr([a75, a71, a15, a11])

其中a455將是一個4x7x5陣列。 理想情況下，我也希望能夠做到

 a2275 = mkarr([[a75, a71], [a15, a11]])

獲得2x2x7x5陣列。

在numpy中表達這些操作最簡潔的方法是什么？

Answer 1

你可以使用np.broadcast_arrays ：

a475 = np.stack(np.broadcast_arrays(a75, a71, a15, a11))

請注意，這會創建原始數組的視圖，因此不應占用任何額外的內存。