【数据分析&数据挖掘】数组的组合&拆分

一、数组的组合

 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)