[[394118]]

1 水平數(shù)組組合

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

現(xiàn)在有兩個(gè)3*2的數(shù)組A和B。

數(shù)組A

0 1 2 3 4 5

數(shù)組B

6 7 8 4 1 5

現(xiàn)在使用hstack函數(shù)將兩個(gè)數(shù)組水平組合的代碼如下。

hstack(A,B) 

hstack函數(shù)的返回值就是組合后的結(jié)果。

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

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

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

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))) 

程序運(yùn)行結(jié)果如圖1所示。

圖1 水平組合數(shù)組

2 垂直數(shù)組組合

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

現(xiàn)在有兩個(gè)3*2的數(shù)組A和B。

數(shù)組A

0 1 2 3 4 5

數(shù)組B

6 7 8 4 1 5

現(xiàn)在使用vstack函數(shù)將兩個(gè)數(shù)組垂直組合的代碼如下。

vstack(A,B) 

vstack函數(shù)的返回值就是組合后的結(jié)果。

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

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

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))) 

程序運(yùn)行結(jié)果如圖2所示。

圖2 垂直組合數(shù)組

本文轉(zhuǎn)載自微信公眾號(hào)「極客起源」，作者geekori。轉(zhuǎn)載本文請聯(lián)系極客起源公眾號(hào)。