1. 程式人生 > >陣列的分割

陣列的分割


import numpy as np

a = np.arange(9).reshape(3,3)
print('原始陣列:\n',a)
b = 2 * a
print('乘2後的陣列:\n',b)

'''
#(1) 水平分割 下面的程式碼將把陣列沿著水平方向分割為3個相同大小的子陣列:
c = np.hsplit(a,3)
print("分割後的陣列:\n",c)
'''

'''
# 對同樣的陣列,呼叫split函式並在引數中指定引數axis=1,對比一下結果:
c = np.split(a,3,axis=1)
print("分割後的陣列:\n",c)
'''

'''
# (2)垂直分割 vsplit函式將把陣列沿著垂直方向分割:
c = np.vsplit(a,3)
print("分割後的陣列:\n",c)

# 同樣,呼叫split函式並在引數中指定引數axis=0,也可以得到同樣的結果:
c = np.split(a,3,axis=0)
print("分割後的陣列:\n",c)
'''

# (3) 深度分割 不出所料,dsplit函式將按深度方向分割陣列。我們先建立一個三維陣列:
c = np.arange(27).reshape(3,3,3)
d = np.dsplit(c,3)
print("分割後的陣列:\n",d)