Python 程序将元素插入到指定索引处的数组中
数组是一种数据结构,用于存储同类数据元素的集合。数组中的每个元素都由索引值标识。
Python 中的数组
Python 没有自己的数据结构来表示数组。但是,我们可以使用列表数据结构作为数组的替代。这里我们将使用列表数组 -
[10, 4, 11, 76, 99]
Python 也提供了一些更合适的模块,这些模块是 Numpy 和数组模块。
NumPy 模块定义的 Numpy 数组是 -
array([1, 2, 3, 4])
python 中的索引从 0 开始,因此可以使用它们各自的索引值(如 0、1、2,直到 n-1)访问上述数组元素。
在下面的文章中,我们看到了使用 Python 编程将元素插入到指定索引处的数组中的多种方法。
输入输出场景
假设我们有一个包含 6 个整数值的数组 A。结果数组将在指定的索引位置插入元素。
输入数组: [9, 3, 6, 2, 7, 1] 输出数组: [9, 3, 6, 2, 10, 7, 1]
元素 10 插入到索引位置 4,插入值后元素数量增加到 7。让我们看看另一个数组。
输入数组: ['a', 'b', 'c'] 输出数组: ['k', 'a', 'b', 'c']
这里元素'k'插入到第一个索引处。
使用list.insert()方法
要将元素插入到指定索引处的数组中,我们可以使用python中的insert()方法。insert()是列表数据结构的内置方法,用于在给定索引处插入元素。以下是此函数的语法 –
list.insert(index, element)
index参数指定插入元素的位置,第二个参数是给定的元素。方法 list.insert() 不返回任何内容,它会更新原始列表。
示例
在此示例中,我们将使用 list.insert() 方法。
l = [9, 3, 6, 2, 7, 1] # 打印初始数组 print("原始数组: ", l) specified_index = 4 element = 10 # 插入元素 l.insert(specified_index, element) print("插入元素后的数组: ", l)
输出
原始数组: [9, 3, 6, 2, 7, 1] 插入元素后的数组: [9, 3, 6, 2, 10, 7, 1]
原始数组 l 在给定位置 4 处更新为新元素 10。
使用 for 循环
无需使用任何方法/函数,我们就可以使用 for 循环将值插入数组中。
示例
在这里我们将使用 python for 循环在特定位置插入值。
l = [3, 6, 2, 8, 4] # 打印初始数组 print("Original array: ", l) specified_index = 1 element = 0 for i in range(len(l)): if i == specified_index: l[i] = element print("插入元素后的数组: ", l)
输出
原始数组:[3, 6, 2, 8, 4] 插入元素后的数组:[3, 0, 2, 8, 4]
这里我们使用数组索引值迭代 for 循环,并将索引与指定索引进行比较,然后插入索引与指定索引匹配的元素。
使用 numpy.insert() 方法
numpy.insert() 方法用于在给定的索引处插入一个值。它返回带有插入值的输入数组的副本。以下是语法 –
numpy.insert(arr, obj, values, axis=None)
参数
arr − 输入 NumPy 数组。
obj − 它定义插入元素的索引。
values − 要插入到输入数组中的元素。
axis − 沿其插入值。默认情况下为 None。
示例
在此示例中,我们将使用 numpy.insert() 方法。
import numpy as np arr = np.array([0, 6, 2, 7, 9, 1, 4]) # 打印初始数组 print("原始数组: ", arr) specified_index = 5 element = 10 # 插入元素 result = np.insert(arr, specified_index, element) print("插入元素后的数组: ", result)
输出
原始数组: [0 6 2 7 9 1 4] 插入元素后的数组: [ 0 6 2 7 9 10 1 4]
使用 numpy.insert() 函数成功将元素 10 插入到位置 5。