[英]Converting 3D matrix to cascaded 2D Matrices
我在python中有一个3D
矩阵,如下所示:
import numpy as np
a = np.ones((2,2,3))
a[0,0,0] = 2
a[0,0,1] = 3
a[0,0,2] = 4
我想将这个3D
矩阵转换为一组2D
矩阵。 我试过np.reshape
但它没有解决我的问题。 我感兴趣的最终形状是以下级联的vesrsion:
[[ 2. 1. 3. 1. 4. 1.]
[ 1. 1. 1. 1. 1. 1.]]
但是, np.reshape
给了我以下内容
[[ 2. 3. 4. 1. 1. 1.]
[ 1. 1. 1. 1. 1. 1.]]
我怎么解决这个问题?
a.transpose([0,2,1]).reshape(a.shape[0],-1)
或者使用与transpose
相同的swapaxes
- reshape
swapaxes
-
a.swapaxes(2,1).reshape(a.shape[0],-1)
样品运行 -
In [66]: a
Out[66]:
array([[[ 2., 3., 4.],
[ 1., 1., 1.]],
[[ 1., 1., 1.],
[ 1., 1., 1.]]])
In [67]: a.transpose([0,2,1]).reshape(a.shape[0],-1)
Out[67]:
array([[ 2., 1., 3., 1., 4., 1.],
[ 1., 1., 1., 1., 1., 1.]])
In [68]: a.swapaxes(2,1).reshape(a.shape[0],-1)
Out[68]:
array([[ 2., 1., 3., 1., 4., 1.],
[ 1., 1., 1., 1., 1., 1.]])
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.