如何使用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水平組合數組和垂直組合數組”有了更深的了解，不妨來實際操作一番吧！這里是億速云網站，更多相關內容可以進入相關頻道進行查詢，關注我們，繼續學習！