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

一、数组的组合

import numpy as np

# 创建数组
arr1 = np.arange(9).reshape((3, 3))
arr2 = np.array([[0, 1, 0],[1, 0, 1],[2, 1, 0]])
print("arr1:
", arr1)
print("arr2:
", arr2)

# 二维拼接 axis=0 行   axis=1 列
res = np.vstack((arr1, arr2))  # 垂直拼接，在行的方向上拼接
res = np.hstack((arr1, arr2))  # 水平拼接，在列的方向上拼接
res = np.concatenate((arr1, arr2), axis=0)  # 在行维度上进行拼接
res = np.concatenate((arr1, arr2), axis=1)  # 在列维度上进行拼接
print("拼接的结果:
", res)

二、数组的拆分

import numpy as np

# 创建数组
arr = np.arange(16).reshape((4, 4))
print("arr:
", arr)

# 拆分数组
res = np.hsplit(arr, 2)  # 水平拆成两部分 按列拆分
res = np.vsplit(arr, 2)  # 垂直拆成两部分 按行拆分
# res = np.vsplit(arr, 3)  # 错误的，3不能被行整除
res = np.split(arr, 2, axis=0)  # 按行拆分
res = np.split(arr, 2, axis=1)  # 按列拆分
print("res:
", res)