博客

numpy中数组的堆叠方法

  • Post category:Python

在NumPy中，可以使用堆叠方法将多个数组沿着不同的轴进行组合。本文将详细讲解NumPy中数组的堆叠方法，包括np.concatenate()函数、np.vstack()函数、np.hstack()函数、np.dstack()函数和np.stack()函数。

np.concatenate()函数

np.concatenate()函数可以将多个数组沿着指定的轴进行拼接。下面是一个示例：

import numpy as np

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

# 沿着行轴拼接数组
c = np.concatenate((a, b), axis=0)

# 打印结果
print(c)

在上面的示例中，我们首先使用np.array()函数创建了两个数组a和b，然后使用np.concatenate()函数将它们沿着行轴拼接成一个新的数组c。最后，print()函数打印出了结果。

需要注意的是，使用np.concatenate()函数拼接数组时，需要指定拼接的轴。如果不指定轴，则默认沿着第0轴（行轴）进行拼接。

np.vstack()函数和np.hstack()函数

np.vstack()函数可以将多个数组沿着垂直方向（行方向）堆叠起来，np.hstack()函数可以将多个数组沿着水平方向（列方向）堆叠起来。下面是两个示例：

示例一：使用np.vstack()函数叠数组

import numpy as np

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

# 沿着垂直方向堆叠数组
c = np.vstack((a, b))

# 打印结果
print(c)

在上面的示例中，首先使用np.array()函数创建了两个数组a和b，然后使用np.vstack()函数将它们沿着垂直方堆叠成一个新的数组c。最后，使用print()函数打印出了结果。

示例二：使用np.hstack()函数堆叠数组

import numpy as np

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

# 沿着水平方向堆叠数组
c = np.hstack((a, b))

# 打印结果
print(c)

在上面的示例中，我们首先使用np.array()函数创建了两个数组`和b，然后使用np.hstack()函数将它们沿着水平方向堆叠成一个新的数组c。最后，使用print()`函数打印出了结果。

np.dstack()函数

np.dstack()函数可以将多个数组沿着深度方向（第三个维度）堆叠起来。下面一个示例：

import numpy as np

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

# 沿着深度方向堆叠数组
c = np.dstack((a, b))

# 打印结果
print(c)

在上面的示例中，我们首先使用np.array()函数创建了两个数组a和b，然后使用np.dstack()函数将它们沿着深度方向堆叠成一个新的数组c。最后，使用print()函数打印出了结果。

np.stack()函数

np()函数可以将多个数组沿着新的轴进行堆叠。下面是一个示例：

import numpy as np

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

# 沿着新的轴堆叠数组
c = np.stack((a, b), axis=2)

# 打印结果
print(c)

在上面的示例中，我们首先使用np.array()函数创建了两个数组a和b，然后使用np.stack()函数将它们沿着新的轴（第三个轴）进行堆叠成一个新数组c。最后，使用print()函数打印出了结果。

需要注意的是，使用np.stack()函数堆叠数组时，需要指定堆叠的轴。

示例一：使用np.concatenate()函数拼接数组

import numpy as np

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

# 沿着列轴拼接数组
c = np.concatenate((a, b), axis=1)

# 打印结果
print(c)

在上面的示例中，我们首先使用np.array()函数创建了两个数组a和b，然后使用np.concatenate()函数将它们沿着列轴拼接成一个新的数组c。最后，使用print()函数打印出了结果。

示例二：使用np.stack()函数堆叠数组

import numpy as np

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

# 沿着新的轴堆叠数组
c = np.stack((a, b), axis=0)

# 打印结果
print(c)

在上面的示例中，我们首先使用np.array()函数创建了两个数组a和b，然后使用np.stack()函数将它们沿着新的轴（第0个轴）进行堆叠成一个新数组c。最后，使用print()函数打印出了结果。

Tags: