【Python科学计算】使用NumPy水平组合数组和垂直组合数组

目录

 

1 水平数组组合

2 垂直数组组合

---

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  垂直组合数组

 

推荐阅读  点击标题可跳转

卧槽，好强大的魔法，竟能让Python支持方法重载

Python装饰器（decorator）不过如此，是我想多了

这样合并Python字典，可以让程序的运行效率提高4倍

Python字典不是不可以排序，是你方法没用对！

文件侠告诉你，Python复制文件的N种姿势！

Python代码可以加密吗？Python字节码告诉你！

使出Python的六脉神剑，让Python拥有无限扩展性

看我用元类（metaclass）花式创建Python类

你不知道__name__变量是什么意思吗？

Python生成器（Generator）最完美解释