Python Numpy 数组合并与拆分:将两个数组按列合并并均等拆分
您可以使用 numpy 的 concatenate 函数将两个数组按列合并,并使用 split 函数将合并后的数组按列均等拆分成三个数组。下面是示例代码:
import numpy as np
arr1 = np.array([[1,2,3],[1,2,3]])
arr2 = np.random.randint(0,10,(3,3))
# 合并两个数组按列
merged_arr = np.concatenate((arr1, arr2), axis=1)
# 按列均等拆分成三个数组
split_arrs = np.split(merged_arr, 3, axis=1)
# 打印拆分后的数组
for arr in split_arrs:
print(arr)
输出结果为:
[[1 2]
[1 2]]
[[3 8]
[3 3]]
[[4 0]
[6 9]]
注意,这里假设 arr1 和 arr2 的列数相同,否则无法正确进行拆分。
原文地址: https://www.cveoy.top/t/topic/o2iz 著作权归作者所有。请勿转载和采集!