我有一个大小为 nxm
的 numpy 数组。我希望将列数限制为 k
并将其余列扩展到新行中。以下是场景 -
初始数组:nxm
最终数组:pxk
其中 p = (m/k)*n
例如。 n = 2,m = 6,k = 2
初始数组:
[[1, 2, 3, 4, 5, 6,],
[7, 8, 9, 10, 11, 12]]
最终数组:
[[1, 2],
[7, 8],
[3, 4],
[9, 10],
[5, 6],
[11, 12]]
我尝试使用 reshape
但没有得到想要的结果。
最佳答案
这里有一个方法可以做到这一点
q=array([[1, 2, 3, 4, 5, 6,],
[7, 8, 9, 10, 11, 12]])
r=q.T.reshape(-1,2,2)
s=r.swapaxes(1,2)
t=s.reshape(-1,2)
作为一个类轮,
q.T.reshape(-1,2,2).swapaxes(1,2).reshape(-1,2)
array([[ 1, 2],
[ 7, 8],
[ 3, 4],
[ 9, 10],
[ 5, 6],
[11, 12]])
编辑:对于一般情况,使用
q=arange(1,1+n*m).reshape(n,m) #example input
r=q.T.reshape(-1,k,n)
s=r.swapaxes(1,2)
t=s.reshape(-1,k)
一个类轮是:
q.T.reshape(-1,k,n).swapaxes(1,2).reshape(-1,k)
n=3,m=12,k=4
的例子
q=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, 33, 34, 35, 36]])
结果是
array([[ 1, 2, 3, 4],
[13, 14, 15, 16],
[25, 26, 27, 28],
[ 5, 6, 7, 8],
[17, 18, 19, 20],
[29, 30, 31, 32],
[ 9, 10, 11, 12],
[21, 22, 23, 24],
[33, 34, 35, 36]])
关于python - 如何切片和扩展二维 numpy 数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/22143352/