在NumPy数组的末尾添加数值

  • Post category:Python

在NumPy中,可以使用numpy.append函数在数组末尾添加数值。下面是使用numpy.append函数添加数值的完整攻略。

函数介绍

numpy.append(arr, values, axis=None)函数将“值”添加到“数组”的末尾,返回一个新的数组。其中,参数arr指定原始数组,values指定要添加到数组末尾的新值或数组。

  • 如果axis无指定或为None,则将输入的数组视为一维数组,返回的数组也为一维数组。

  • 如果axis有指定值,则输入的数组视为多维数组,返回的数组也将是多维数组。

使用示例

示例1:在一维数组末尾添加一个值

import numpy as np

# 创建一个一维数组
a = np.array([1, 2, 3, 4])
print("原始数组:", a)

# 添加一个值到数组末尾
b = np.append(a, 5)
print("添加值后的新数组:", b)

输出结果如下:

原始数组: [1 2 3 4]
添加值后的新数组: [1 2 3 4 5]

示例2:在二维数组末尾添加一行数据

import numpy as np

# 创建一个二维数组
c = np.array([[1, 2], [3, 4]])
print("原始数组:\n", c)

# 添加一行数据到数组末尾
d = np.append(c, [[5, 6]], axis=0)
print("添加值后的新数组:\n", d)

输出结果如下:

原始数组:
 [[1 2]
 [3 4]]
添加值后的新数组:
 [[1 2]
  [3 4]
  [5 6]]

在本示例中,axis=0表示在第一个轴(即行)上添加新值,因此添加的新值需要使用二维数组 [5,6] 包装。