我有 4 个数组,我想将它们连接成一个交错的数组。我该怎么做?
>>> import numpy as np
>>> a = np.tile(0,(5,2))
>>> b = np.tile(1,(5,2))
>>> c = np.tile(2,(5,2))
>>> d = np.tile(3,(5,2))
>>> e = np.concatenate((a,b,c,d),axis=1)
>>> e
array([[0, 0, 1, 1, 2, 2, 3, 3],
[0, 0, 1, 1, 2, 2, 3, 3],
[0, 0, 1, 1, 2, 2, 3, 3],
[0, 0, 1, 1, 2, 2, 3, 3],
[0, 0, 1, 1, 2, 2, 3, 3]])
这仅给出串联。
但是,我想要的输出是:
>>> desired_output
array([[0, 1, 2, 3, 0, 1, 2, 3],
[0, 1, 2, 3, 0, 1, 2, 3],
[0, 1, 2, 3, 0, 1, 2, 3],
[0, 1, 2, 3, 0, 1, 2, 3],
[0, 1, 2, 3, 0, 1, 2, 3]])
我的意思是我知道我可以使用以下方法从 e 中选择交错列:
>>> f = e[:, ::2]
>>> array([[0, 1, 2, 3],
[0, 1, 2, 3],
[0, 1, 2, 3],
[0, 1, 2, 3],
[0, 1, 2, 3]])
但是如何制作一个大数组呢?
最佳答案
使用np.dstack
或np.stack
沿着最后一个轴堆叠,得到一个 3D
数组,然后重新整形回 2D
-
np.dstack([a,b,c,d]).reshape(a.shape[0],-1)
np.stack([a,b,c,d],axis=2).reshape(a.shape[0],-1)
关于python - Numpy 通过交错连接数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47430784/