[[394118]]

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 垂直組合數組

本文轉載自微信公眾號「極客起源」，作者geekori。轉載本文請聯系極客起源公眾號。