python数组拆分_Python在机器学习中切割与合并数组的几个函数

在人工智能领域，特别是在深度学习中会经常遇到数组需要切割或者拼接，python中处理数组分割与拼接的方法很多，本文将介绍使用numpy处理数组拼接与分割的几个函数。

1.数组的分割：均等分割-numpy.split()

split(ary, indices_or_sections, axis=0)

参数：

ary：要切分的数组

indices_or_sections：如果是一个整数，就用该数平均切分，如果是一个数组，为沿轴切分的位置（左开右闭）

axis：沿着哪个维度进行切向，默认为0，横向切分。为1时，纵向切分

注意：如果根据提供的参数不能实现均等切分，会报错。

# 1. 生成测试数组

A = np.arange(24).reshape(6,4)

print(A)

# 2. 分割数组

# 2.1 均等切割，如果不能均等切割则会报错

split_A1, split_A2 = np.split(A,2,axis=0) # split均等切割，如果不能均等切割，这则会报错

print("split_A1:\n", split_A1, "\n", "split_A2:\n", split_A2)

运行结果：

2.非均等分割-numpy.array_split()

参数：

ary：要切分的数组

indices_or_sections：如果是一个整数，就用该数