import numpy as np
a = np.array([[1,2],
[3,4],
[5,6],
[7,8],
[9,10],
[11,12]])
print np.shape(a)
预期的答案应该是:
answer = np.array([[1,2,7,8],
[3,4, 9, 10],
[5,6, 11, 12]])
我尝试了
ans = a.reshape(3,-1)
print ans
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]
但是答案是错误的。怎么做?
我将使用split
此操作:
In [110]: np.hstack(np.split(a,2))
Out[110]:
array([[ 1, 2, 7, 8],
[ 3, 4, 9, 10],
[ 5, 6, 11, 12]])
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句