Python 程序将元素插入到指定索引处的数组中

pythonserver side programmingprogramming

数组是一种数据结构,用于存储同类数据元素的集合。数组中的每个元素都由索引值标识。

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。


相关文章