在NumPy数组的末尾添加数值

  • Post category:Python

在NumPy数组的末尾添加数值,可以使用NumPy提供的np.append()函数来实现。下面详细讲解该函数的使用方法:

函数原型

np.append(arr, values, axis=None)

参数说明

  • arr:被添加的数组
  • values:要添加到数组中的值,可以是单个数值,也可以是数组。如果传入数组,那么该数组的形状(Shape)必须与arr的形状相同(除了axis维的长度之外)
  • axis:指定在哪个维度上添加数值,如果不指定axis,则会将输入的数组展开,再添加新的数值。

示例说明

示例一

import numpy as np

# 定义一个3*3的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 在末尾添加一个数值
arr_new = np.append(arr, 10)

print("原数组:")
print(arr)

print("添加新值后的数组:")
print(arr_new)

输出结果:

原数组:
[[1 2 3]
 [4 5 6]
 [7 8 9]]
添加新值后的数组:
[ 1  2  3  4  5  6  7  8  9 10]

示例二

import numpy as np

# 定义一个2*2*2的三维数组
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

# 在第0个维度上添加一个形状为(2, 2)的二维数组
new_values = np.array([[9, 10], [11, 12]])
arr_new = np.append(arr, new_values, axis=0)

print("原数组:")
print(arr)

print("添加新值后的数组:")
print(arr_new)

输出结果:

原数组:
[[[1 2]
  [3 4]]

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

 [[ 5  6]
  [ 7  8]]

 [[ 9 10]
  [11 12]]]

以上便是在NumPy数组末尾添加数值的完整攻略,可以根据需要选择相应的示例进行参考。