怎么使用NumPy實現水平組合數組和垂直組合數組

這篇文章主要介紹了怎么使用NumPy實現水平組合數組和垂直組合數組的相關知識，內容詳細易懂，操作簡單快捷，具有一定借鑒價值，相信大家閱讀完這篇怎么使用NumPy實現水平組合數組和垂直組合數組文章都會有所收獲，下面我們一起來看看吧。

1 水平數組組合

通過hstack函數可以將2個或多個數組水平組合起來形成一個數組，那么什么叫數組的水平組合呢?下面先看一個例子。

現在有兩個3*2的數組A和B。

數組A

0 1 2 3 4 5

數組B

6 7 8 4 1 5

現在使用hstack函數將兩個數組水平組合的代碼如下。

hstack(A,B)

hstack函數的返回值就是組合后的結果。

0 1 2 6 7 8 3 4 5 4 1 5

我們可以看到，數組A和數字B在水平方向首尾連接了起來，形成了一個新的數組。這就是數組的水平組合。多個數組進行水平組合的效果類似。但數組水平組合必須要滿足一個條件，就是所有參與水平組合的數組的行數必須相同，否則進行水平組合會拋出異常。

下面的例子通過reshape方法以及乘法運行創建了3個二維數組(行數相同)，然后使用hstack函數水平組合其中的兩個或三個數組。

from numpy import * a = arange(9).reshape(3,3) b = a * 3 print(a) print('----------------') print(b) print('----------------') c = a * 5 # 水平組合a和b print(hstack((a,b))) print('----------------') # 水平組合a、b和c print(hstack((a,b,c)))

程序運行結果如圖1所示。

圖1 水平組合數組

2 垂直數組組合

通過vstack函數可以將2個或多個數組垂直組合起來形成一個數組，那么什么叫數組的垂直組合呢?下面先看一個例子。

現在有兩個3*2的數組A和B。

數組A

0 1 2 3 4 5

數組B

6 7 8 4 1 5

現在使用vstack函數將兩個數組垂直組合的代碼如下。

vstack(A,B)

vstack函數的返回值就是組合后的結果。

0 1 2 3 4 5 6 7 8 4 1 5

下面的例子通過reshape方法以及乘法運行創建了3個二維數組(行數相同)，然后使用hstack函數水平組合其中的兩個或三個數組。

from numpy import * a = arange(12).reshape(3,4) b = arange(16).reshape(4,4) c = arange(20).reshape(5,4) print(a) print('------------') print(b) print('------------') print(c) print('------------') # 垂直組合 print(vstack((a,b,c)))

程序運行結果如圖2所示。

圖2 垂直組合數組

關于“怎么使用NumPy實現水平組合數組和垂直組合數組”這篇文章的內容就介紹到這里，感謝各位的閱讀！相信大家對“怎么使用NumPy實現水平組合數組和垂直組合數組”知識都有一定的了解，大家如果還想學習更多知識，歡迎關注億速云行業資訊頻道。