NumPy | 数组组合与分割

>>> a = arange(8).reshape(2, 4)
//数组可直接运算
>>> b = a * 4
>>> a, b
(array([[0, 1, 2, 3],
       [4, 5, 6, 7]]), array([[ 0,  4,  8, 12],
       [16, 20, 24, 28]]))

//水平组合
>>> hstack((a, b))
array([[ 0,  1,  2,  3,  0,  4,  8, 12],
       [ 4,  5,  6,  7, 16, 20, 24, 28]])
//垂直组合
>>> vstack((a, b))
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 0,  4,  8, 12],
       [16, 20, 24, 28]])
//深度组合
>>> dstack((a, b))
array([[[ 0,  0],
        [ 1,  4],
        [ 2,  8],
        [ 3, 12]],

       [[ 4, 16],
        [ 5, 20],
        [ 6, 24],
        [ 7, 28]]])

//水平分割
>>> hsplit(a, 2)
[array([[0, 1],
       [4, 5]]), array([[2, 3],
       [6, 7]])]
>>> hsplit(a, 4)
[array([[0],
       [4]]), array([[1],
       [5]]), array([[2],
       [6]]), array([[3],
       [7]])]
//垂直分割
>>> vsplit(a, 2)
[array([[0, 1, 2, 3]]), array([[4, 5, 6, 7]])]