我想將具有不同但廣播兼容的數組的數組堆疊在一起。給定 7x5、7x1、1x5 和 1x1 數組，我想做一些類似的事情 a475 = mkarr([a75, a71, a15, a11])a4554x7x5 陣列在哪里。理想情況下，我也希望能夠做到 a2275 = mkarr([[a75, a71], [a15, a11]])得到一個 2x2x7x5 的數組。在 numpy 中表達這些操作的最簡潔方法是什么？
1 回答
qq_花開花謝_0
TA貢獻1835條經驗 獲得超7個贊
您可以使用np.broadcast_arrays：
a475 = np.stack(np.broadcast_arrays(a75, a71, a15, a11))
請注意，這會創建原始數組的視圖，因此它不應需要任何額外的內存使用。
添加回答
舉報
0/150
提交
取消