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。转载本文请联系极客起源公众号。