问题
假定有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个原始的一维数组中依次分别取一个元素进行交叉拼接后的结果。
解决--zip()函数
先将原始的三个一维数组x[0]、x[1]、x[2]使用zip函数进行打包,得到包含三个元组的列表,即([(1, 4, 7), (2, 5, 8), (3, 6, 9)]),然后使用numpy库中的函数将其转化为3×3的二维数组,最后将这个二维数组使用flatten函数转换为最终的拼接数组y即可。
y = np.array(zip(x0, x1, x2)).flatten()
不加上flatten()的话,就是3个一维数组元素交叉拼接的二维数组了。
参考链接:https://baijiahao.baidu.com/s?id=1630485218560231954&wfr=spider&for=pc