numpy中的hstack()、vstack()、stack()、concatenate()函数详解

目录

1. concatenate()

2. stack()

3. vstack()

4. hstack() 

5. tf中的stack()

---

本文主要介绍一下numpy中的几个常用函数，包括hstack()、vstack()、stack()、concatenate()。

1. concatenate()

我们先来介绍最全能的concatenate()函数，后面的几个函数其实都可以用concatenate()函数来进行等价操作。

concatenate()函数根据指定的维度，对一个元组、列表中的list或者ndarray进行连接，函数原型：

numpy.concatenate((a1, a2, ...), axis=0)

先来看几个例子，一个2*2的数组和一个1*2的数组，在第0维进行拼接，得到一个3*2的数组：

import numpy as np
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6]])
np.concatenate((a, b), axis=0)

 输出为：

array([[1, 2],
       [3, 4],
       [5, 6]])

进一步，一个2*2的数组和一个2*1的数组，在第01维进行拼接，得到一个2*3的数组：

np.concatenate((a, b.T), axis=1)

输出为：