问题提出
假定有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,第一种方法更好理解,第二种方法在代码实现上则更为简单。本文完整的测试源代码如下:
运行结果如下:
从结果中也可以看出,两种方案都能得到正确的结果。
本文由编码那些事原创,请关注+转发+收藏+点赞,带你一起长知识!