在NumPy数组的末尾添加数值

  • Post category:Python

要在NumPy数组的末尾添加数值,可以使用numpy.append()函数。该函数接受三个参数:

  • arr:要附加到其中的输入数组
  • values:要附加到 arr 末尾的值或数组
  • axis:用于沿其附加的轴。默认为 None,表示输入数组将被展开。如果给定了轴,则必须指定一个长度为1的值或数组。

下面是一个完整的攻略,它演示了如何在NumPy数组的末尾添加值:

  1. 导入NumPy模块:
import numpy as np
  1. 创建一个NumPy数组:
arr = np.array([1, 2, 3])
  1. 使用numpy.append()函数在数组末尾添加一个单独的值:
new_arr = np.append(arr, 4)
print(new_arr)  # 输出 [1 2 3 4]
  1. 使用numpy.append()函数在数组末尾添加多个值:
new_arr = np.append(arr, [4, 5, 6])
print(new_arr)  # 输出 [1 2 3 4 5 6]

下面是另一个示例,它展示了如何将一个二维NumPy数组附加到另一个二维NumPy数组的末尾:

# 创建两个二维NumPy数组
arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6]])

# 附加 arr2 到 arr1 的末尾
new_arr = np.append(arr1, arr2, axis=0)
print(new_arr)
# 输出
# [[1 2]
#  [3 4]
#  [5 6]]

在此示例中,我们通过指定axis=0参数告诉numpy.append()函数将arr2添加到arr1的末尾,形成一个新的二维数组。请注意,我们需要将arr2转换为二维数组,以便可以沿指定的轴附加它。