如何追加两个NumPy数组

  • Post category:Python

下面我来为你详细讲解如何追加两个NumPy数组。

使用numpy.concatenate()函数

numpy.concatenate()是NumPy中用于连接两个或多个数组的函数。它包含以下几个参数:

numpy.concatenate((a1, a2, ...), axis=0, out=None)
  • a1, a2, ...: 要连接的数组序列。
  • axis: 沿着哪个轴连接数组,默认为0
  • out: 可选参数,输出结果存放的位置。

下面是两个示例说明:

示例1:横向拼接两个数组

import numpy as np

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

# 横向拼接两个数组
result = np.concatenate((a1, a2), axis=1)

print(result)

输出结果为:

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

示例2:纵向拼接两个数组

import numpy as np

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

# 纵向拼接两个数组
result = np.concatenate((a1, a2), axis=0)

print(result)

输出结果为:

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

使用numpy.concatenate()函数可以方便地追加两个NumPy数组。