NumPy 数组组合分割

原文地址：http://blog.csdn.net/lsjseu/article/details/20359201

在原文基础上适当精简并更正少量原作者的笔误:

 

1. 组合(stack)不同的数组

几种方法可以沿不同轴将数组堆叠在一起：

>>> a =floor(10*random.random((2,2)))>>> aarray([[ 1.,  1.],       [ 5., 8.]])>>> b =floor(10*random.random((2,2)))>>> barray([[ 3.,  3.],       [ 6., 0.]])>>> vstack((a,b)) #verticalarray([[ 1.,  1.],       [ 5., 8.],       [ 3., 3.],       [ 6., 0.]])>>> hstack((a,b)) #horizonalarray([[ 1.,  1., 3.,  3.],       [ 5., 8.,  6.,  0.]])

2. 函数 column_stack 以列将一维数组合成二维数组，它等同与vstack 对一维数组:

>>>column_stack((a,b))   # With 2D arraysarray([[ 1.,  1., 3.,  3.],       [ 5., 8.,  6.,  0.]])>>>a=array([4.,2.])>>>b=array([2.,8.])>>>a[:,newaxis] # This allows tohave a 2D columns vectorarray([[ 4.],       [ 2.]])>>>column_stack((a[:,newaxis],b[:,newaxis]))array([[ 4.,  2.],       [ 2., 8.]])>>>vstack((a[:,newaxis],b[:,newaxis]))# The behavior ofvstack is differentarray([[ 4.],       [ 2.],       [ 2.],       [ 8.]])

在复杂情况下， r_[] 和c_[] 对创建沿着一个方向组合的数很有用，它们允许范围符号(“:”):

>>> r_[1:4,0,4]array([1, 2, 3, 0,4])

3. 将一个数组分割(split)成几个小数组:

使用 hsplit你能将数组进行纵向分割，或者指定返回相同形状数组的个数，或者指定在哪些列后发生分割:

>>> a =floor(10*random.random((2,12)))>>> aarray([[ 8., 8., 3.,  9.,  0.,   4.,  3.,  0.,   0.,  6.,  4.,   4.],     [ 0., 3., 2.,  9.,   6.,  0.,  4.,   5.,  7.,  5.,   1.,  4.]])>>> hsplit(a,3)   # Split a into 3[array([ [ 8., 8.,   3.,  9.],      [ 0., 3.,  2.,  9.]]),  array([[ 0., 4.,   3.,  0.],      [ 6., 0.,  4.,  5.]]),  array([[ 0., 6.,   4.,  4.],      [ 7., 5.,  1.,  4.]])]>>>hsplit(a,(3,4))   # Split a after thethird and the fourth column[array([ [ 8., 8., 3.],      [ 0., 3., 2.]]),  array([[ 9.],      [ 9.]]),  array([[ 0.,  4.,   3.,  0.,  0.,   6.,  4.,  4.],      [ 6.,   0.,  4.,  5.,   7.,  5.,  1.,   4.]])]

vsplit降数组进行横向分割，arraysplit允许指定沿哪个轴分割。