Python 将一个元素插入到指定索引的数组中

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

Python中的数组

Python没有自己的数据结构来表示数组。然而，我们可以使用列表数据结构作为数组的替代。在这里，我们将使用列表作为数组-

[10, 4, 11, 76, 99]

Python还提供了一些更合适的模块，这些模块是NumPy和array模块。

由NumPy模块定义的NumPy数组为 –

array([1, 2, 3, 4])

在Python中，索引从0开始，因此上述数组元素通过它们各自的索引值访问，如0、1、2、直至n-1。

在下面的文章中，我们将看到使用Python编程将元素插入到指定索引位置的多种方法。

输入输出情景

假设我们有一个包含6个整数值的数组A，而结果数组将在指定的索引位置插入元素。

Input array:
[9, 3, 6, 2, 7, 1]
Output array:
[9, 3, 6, 2, 10, 7, 1]

在索引位置4插入元素10，并且插入值后元素计数增加到7。让我们看看另一个数组。

Input arrays:
[‘a’, ‘b’, ‘c’]
Output array:
[‘k’, ‘a’, ‘b’, ‘c’]

在第一个索引处插入元素’k’。 使用list.insert()方法 要在指定的索引处将元素插入数组中，我们可以使用Python中的insert()方法。insert()是一个列表数据结构的内置方法，用于在给定的索引处插入元素。以下是该函数的语法 –

list.insert(index, element)

索引参数指定要插入元素的位置，第二个参数是给定的元素。 list.insert() 方法不返回任何内容，它只是更新原始列表。

示例

在这个示例中，我们将使用 list.insert() 方法。

l = [9, 3, 6, 2, 7, 1]
# print initial array
print("Original array: ", l)

specified_index = 4
element = 10

#  insert element
l.insert(specified_index, element)

print("Array after inserting the element: ", l)

输出

Original array:  [9, 3, 6, 2, 7, 1]
Array after inserting the element:  [9, 3, 6, 2, 10, 7, 1]

原始数组l在给定的位置4更新了新元素10。

使用for循环

在不使用任何方法/函数的情况下，我们可以使用for循环将一个值插入到数组中。

示例

在这里，我们将使用python的for循环在特定位置插入一个值。

l = [3, 6, 2, 8, 4]
# print initial array
print("Original array: ", l)
specified_index = 1
element = 0

for i in range(len(l)):
   if i == specified_index:
      l[i] = element

print("Array after inserting the element: ", l)

输出

Original array:  [3, 6, 2, 8, 4]
Array after inserting the element:  [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 initial array
print("Original array: ", arr)

specified_index = 5
element = 10

#  insert element
result = np.insert(arr, specified_index, element)

print("Array after inserting the element: ", result)

输出

Original array:  [0 6 2 7 9 1 4]
Array after inserting the element:  [ 0  6  2  7  9 10  1  4]

使用numpy.insert()函数，在位置5成功插入元素10。