[英]numpy array construction with broadcasting
我想将具有不同但广播兼容阵列的阵列堆叠在一起。 给定7x5,7x1,1x5和1x1阵列,我想做类似的事情
a475 = mkarr([a75, a71, a15, a11])
其中a455
将是一个4x7x5阵列。 理想情况下,我也希望能够做到
a2275 = mkarr([[a75, a71], [a15, a11]])
获得2x2x7x5阵列。
在numpy中表达这些操作最简洁的方法是什么?
你可以使用np.broadcast_arrays
:
a475 = np.stack(np.broadcast_arrays(a75, a71, a15, a11))
请注意,这会创建原始数组的视图,因此不应占用任何额外的内存。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.