问题提出

假定有3个一维数组x0、x1、x2，其元素分别为:

x0 = [1, 2, 3]

x1 = [4, 5, 6]

x2 = [7, 8, 9]

请将这3个一维数组的元素交叉拼接后，组成一个新的一维数组y:

y = [1, 4, 7, 2, 5, 8, 3, 6, 9]

即新的数组y是从3个原始的一维数组中依次分别取一个元素进行交叉拼接后的结果。

在Python中如何编程实现呢？

先仿真3个一维数组，代码如下图所示：

上面代码中使用numpy库中的arange函数产生了三个一维数组，每个数组包含三个元素，将这三个一维数组放在一个字典x中，分别对应为x[0]、x[1]、x[2]。

下面给出得到拼接数组y的两种方法，仅供参考。

方法1

由于拼接后的数组包含9个元素，可以使用numpy库中的zeros函数先产生一个包含9个元素的初始化数组y1，其每个值为0，然后在for循环中将这三个一维数组的元素每隔3个元素，分别赋值（替换）到对应的位置，代码如下：

方法2

这种方法更为简单，先将原始的三个一维数组x[0]、x[1]、x[2]使用zip函数进行打包，得到包含三个元组的列表，即（[(1, 4, 7), (2, 5, 8), (3, 6, 9)]），然后使用numpy库中的函数将其转化为3×3的二维数组，最后将这个二维数组使用flatten函数转换为最终的拼接数组y即可。实现代码如下：

结论

上面两种方法都可以得到拼接后的数组y，第一种方法更好理解，第二种方法在代码实现上则更为简单。本文完整的测试源代码如下：

运行结果如下：

从结果中也可以看出，两种方案都能得到正确的结果。

本文由编码那些事原创，请关注+转发+收藏+点赞，带你一起长知识！