数组的合并

#coding=utf-8
import numpy as np

'''
数组的合并：
np.concatnate([a,b],axis)
axis=0  表示垂直方向堆叠数组，必须保持两数组的列数一致
axis=1  表示水平方向堆叠数组，必须保持两数组的行数一致
'''
a=np.arange(6).reshape(2,3)
b=np.arange(3).reshape(1,3)
c=np.arange(10).reshape(2,5)

print(a)
print(b)
print(c)
'''
[[0 1 2]
 [3 4 5]]
 
 [[0 1 2]]
 
 [[0 1 2 3 4]
 [5 6 7 8 9]]
'''
#垂直方向堆叠数组，必须保持两数组的列数一致
print(np.concatenate([a,b],axis=0))
'''
[[0 1 2]
 [3 4 5]
 [0 1 2]]
 
print(np.concatenate([a,c],axis=0))
报错，因为两数组的列数不一致
ValueError: all the input array dimensions except for the concatenation axis must match exactly
'''
#垂直方向堆叠数组，必须保持两数组的列数一致，方式二
print(np.vstack([a,b]))
'''
[[0 1 2]
 [3 4 5]
 [0 1 2]]
'''


#水平方向堆叠数组，必须保持两数组的行数一致
print(np.concatenate([a,c],axis=1))
'''
[[0 1 2 0 1 2 3 4]
 [3 4 5 5 6 7 8 9]]
 
print(np.concatenate([b,c],axis=1))
报错，因为两数组的行数不一致
ValueError: all the input array dimensions except for the concatenation axis must match exactly 
'''
#水平方向堆叠数组，必须保持两数组的行数一致，方式二
print(np.hstack([a,c]))
'''
[[0 1 2 0 1 2 3 4]
 [3 4 5 5 6 7 8 9]]
'''

数组的分割

'''
数组的分割：
np.split(ary, indices_or_sections, axis=0)
ary：   表示数组
axis=0：表示水平分割(默认的)，axis=1：表示垂直分割
indices_or_sections=：
可以是整数也可以是区间
=N整数，表示分割成N段，如果N无法整除，则会报错
=[2,5]区间，表示分割成ary[:2]  ary[2:5]  ary[5:]
'''
a=np.arange(9)
print(np.split(a,3))
'''
[array([0, 1, 2]), array([3, 4, 5]), array([6, 7, 8])]
print(np.split(a,2))
报错：
ValueError: array split does not result in an equal division
'''

print(np.split(a,[2,5]))
'''
[array([0, 1]), array([2, 3, 4]), array([5, 6, 7, 8])]
'''

b=np.arange(16).reshape(4,4)
print(b)
print(np.split(b,2,axis=0))
记：因为axis=0是垂直方向堆叠，分割时自然也是对着垂直方向九十度分割
'''
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]
 [12 13 14 15]]
 
 
[array([[0, 1, 2, 3],
       [4, 5, 6, 7]]),
array([[ 8,  9, 10, 11],
       [12, 13, 14, 15]])]
'''

print(np.split(b,2,axis=1))
记：因为axis=1是水平方向堆叠，分割时自然也是对着水平方向九十度分割
'''
[array([[ 0,  1],
       [ 4,  5],
       [ 8,  9],
       [12, 13]]), 
array([[ 2,  3],
       [ 6,  7],
       [10, 11],
       [14, 15]])]
'''