python使用numpy合并两个或多个数组

2023-03-10 00:00:00 多个 合并 数组

在NumPy中,可以使用numpy.concatenate()函数来合并两个或多个数组。该函数需要传入一个元组或列表,其中包含要合并的数组。
下面是使用numpy.concatenate()函数合并两个数组的示例代码:

import numpy as np
# 创建两个数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6]])
# 沿着行的方向合并两个数组
c = np.concatenate((a, b), axis=0)
# 输出合并后的数组
print(c)

输出结果为:

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

在上面的代码中,a和b是要合并的两个数组,使用numpy.concatenate()函数将它们沿着行的方向合并成了一个新的数组c。需要注意的是,传入axis=0表示沿着行的方向进行合并。
除了numpy.concatenate()函数,还可以使用numpy.vstack()和numpy.hstack()函数分别在垂直和水平方向合并两个或多个数组。例如:

# 在垂直方向合并两个数组
c = np.vstack((a, b))
# 在水平方向合并两个数组
c = np.hstack((a, b))

这些函数的参数和numpy.concatenate()函数相同。使用它们可以更方便地合并数组。

相关文章