我有一个数组:
> a
array([[[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12],
[13, 14, 15, 16]],
[[17, 18, 19, 20],
[21, 22, 23, 24],
[25, 26, 27, 28],
[29, 30, 31, 32]]])
我如何将其重塑为:
[[1, 2, 3, 4, 17, 18, 19, 20],
[5, 6, 7, 8, 21, 22, 23, 24],
[9, 10, 11, 12, 25, 26, 27, 28],
[13, 14, 15, 16, 29, 30, 31, 32]]
我尝试了reshape
函数,但它返回如下:
[[1, 2, 3, 4, 5, 6, 7, 8],
[9, 10, 11, 12, 13, 14, 15, 16],
[17, 18, 19, 20, 21, 22, 23, 24],
[25, 26, 27, 28, 29, 30, 31, 32]]
有什么解决办法吗?
非常感谢
3条答案
按热度按时间zf9nrax11#
Numpy通过hstack方法内置了此功能
tvmytwxo2#
您可以使用
moveaxis
将第一个轴推到最后,然后按Fortran顺序使用reshape
:输出量:
klr1opcd3#
可以使用
np.c_