Python中的np.vstack()和np.hstack()详解

Python中的np.vstack()和np.hstack()详解

在Python的科学计算库NumPy中，我们有两个非常重要的函数：np.vstack()和np.hstack()，它们可以用来合并数组。下面我们详细阐述这两个函数的用法。

np.vstack()

np.vstack()是一个用于垂直堆叠（vertically stack）数组的函数。具体来说，它可以将两个或多个NumPy数组沿着垂直方向堆叠成一个数组，其实就是把数组按行方向堆叠起来。

下面是一个例子：

import numpy as np

# 定义两个数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6]])

# 使用np.vstack()合并数组
c = np.vstack((a, b))

# 输出结果
print(c)

运行结果：

[[1 2]
 [3 4]
 [5 6]]

在这个例子中，我们定义了两个数组a和b，其中a是一个2x2的数组，b是一个1x2的数组。使用np.vstack()函数，我们将这两个数组沿着垂直方向堆叠成了一个3x2的数组c，其中a的内容在前，b的内容在后。

np.hstack()

np.hstack()是一个用于水平堆叠（horizontally stack）数组的函数。与np.vstack()类似，它可以将两个或多个NumPy数组沿着水平方向堆叠成一个数组，其实就是把数组按列方向堆叠起来。

下面是一个例子：

import numpy as np

# 定义两个数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

# 使用np.hstack()合并数组
c = np.hstack((a, b))

# 输出结果
print(c)

运行结果：

[[1 2 5 6]
 [3 4 7 8]]

在这个例子中，我们定义了两个数组a和b，它们都是2x2的数组。使用np.hstack()函数，我们将这两个数组沿着水平方向堆叠成了一个2x4的数组c，其中a的内容在前两列，b的内容在后两列。

希望这个“Python中的np.vstack()和np.hstack()详解”的攻略可以帮助到你。