Numpy中stack(),hstack(),vstack()函数用法介绍及实例

  • Post category:Python

下面是关于“Numpy中stack(),hstack(),vstack()函数用法介绍及实例”的完整攻略,包含了两个示例。

stack()函数

stack()函数是Numpy中用于沿着新轴连接数组序列的函数。下面是一个示例,演示如何使用stack()函数将两个一维数组沿着新轴连接成一个二维数组。

import numpy as np

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

# 使用stack()函数将两个一维数组沿着新轴连接成一个二维数组
c = np.stack((a, b), axis=1)

print(c)

在上面的示例中,我们首先创建了两个一维数组a和b,然后使用stack()函数将它们沿着新轴连接成一个二维数组c。输出结果为:

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

需要注意的是,stack()函数的axis参数指定了新轴的位置。在上面的示例中,我们使用axis=1将两个一维数组沿着新轴连接成一个二维数组。

hstack()函数

hstack()函数是Numpy中用于沿着水平方向连接数组序列的函数。下面是一个示例,演示如何使用hstack()函数将两个一维数组沿着水平方向连接成一个一维数组。

import numpy as np

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

# 使用hstack()函数将两个一维数组沿着水平方向连接成一个一维数组
c = np.hstack((a, b))

print(c)

在上面的示例中,我们首先创建了两个一维数组a和b,然后使用hstack()函数将它们沿着水平方向连接成一个一维数组c。输出结果为:

[1 2 3 4 5 6]

需要注意的是,hstack()函数只用于连接一维数组,如果要连接多维数组,可以使用stack()函数。

vstack()函数

vstack()函数是Numpy中用于沿着垂直方向连接数组序列的函数。下面是一个示例,演示如何使用vstack()函数将两个一维数组沿着垂直方向连接成一个二维数组。

import numpy as np

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

# 使用vstack()函数将两个一维数组沿着垂直方向连接成一个二维数组
c = np.vstack((a, b))

print(c)

在上面的示例中,我们首先创建了两个一维数组a和b,然后使用vstack()将它们沿着垂直方向连接成一个二维数组c。输出结果为:

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

需要注意的是,vstack()函数只能用于连接一维数组,如果要连接多维数组,可以使用stack()函数。