在我们编写Python程序时,有时需要将新的元素动态添加到数组中。对于这个问题,NumPy提供了多种方法来解决。本文将介绍向NumPy数组中添加元素的不同方法。
1、将元素添加到末尾
最简单的方法是将元素添加到现有数组的末尾。可以使用NumPy库中的append()
函数来实现这一点。函数需要两个参数——要添加的后缀和原始数组。
import numpy as np
# 创建一个长度为5的整数数组
arr = np.array([1, 2, 3, 4, 5])
print("原始数组:", arr)
# 使用append()函数将6附加到数组的末尾
new_arr = np.append(arr, 6)
print("新数组:", new_arr)
输出:
原始数组: [1 2 3 4 5]
新数组: [1 2 3 4 5 6]
2、将元素添加到指定位置
如果想要将元素添加到已有数组的特定位置,可以使用insert()
函数。函数需要三个参数——原始数组、要插入的索引位置和要插入的值