一、数组的组合
1 import numpy as np 2 3 # 创建数组 4 arr1 = np.arange(9).reshape((3, 3)) 5 arr2 = np.array([[0, 1, 0],[1, 0, 1],[2, 1, 0]]) 6 print("arr1: ", arr1) 7 print("arr2: ", arr2) 8 9 # 二维拼接 axis=0 行 axis=1 列 10 res = np.vstack((arr1, arr2)) # 垂直拼接，在行的方向上拼接 11 res = np.hstack((arr1, arr2)) # 水平拼接，在列的方向上拼接 12 res = np.concatenate((arr1, arr2), axis=0) # 在行维度上进行拼接 13 res = np.concatenate((arr1, arr2), axis=1) # 在列维度上进行拼接 14 print("拼接的结果: ", res)
二、数组的拆分
1 import numpy as np 2 3 # 创建数组 4 arr = np.arange(16).reshape((4, 4)) 5 print("arr: ", arr) 6 7 # 拆分数组 8 res = np.hsplit(arr, 2) # 水平拆成两部分 按列拆分 9 res = np.vsplit(arr, 2) # 垂直拆成两部分 按行拆分 10 # res = np.vsplit(arr, 3) # 错误的，3不能被行整除 11 res = np.split(arr, 2, axis=0) # 按行拆分 12 res = np.split(arr, 2, axis=1) # 按列拆分 13 print("res: ", res)