Python程序移除数组中前几个元素


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

Python中的数组

Python本身没有原生数组数据结构。相反,我们可以使用列表数据结构来表示数组。

[1, 2, 3, 4, 5]

我们还可以使用array或NumPy模块来处理Python中的数组。由array模块定义的数组为:

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

NumPy模块定义的NumPy数组为:

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

Python索引从0开始。上面所有数组的索引都从0开始到(n-1)。

输入输出场景

假设我们有一个包含5个元素的整数数组。在输出数组中,将移除前几个元素。

Input array:
[1, 2, 3, 4, 5]
Output:
[3, 4, 5]

从输入数组中移除前两个元素1、2。

在本文中,我们将了解如何从数组中移除前几个给定数量的元素。这里我们主要使用Python切片来移除元素。

Python中的切片

切片允许一次访问多个元素,而不是使用索引访问单个元素。

语法

iterable_obj[start:stop:step]

其中,

  • 开始:切片对象开始的起始索引。默认值为0。

  • 结束:切片对象停止的结束索引。默认值为len(object)-1。

  • 步长:起始索引的增量值。默认值为1。

使用列表

我们可以使用列表切片从数组中移除前几个给定数量的元素。

示例

让我们举个例子,并应用列表切片从数组中移除前几个元素。

# creating array
lst = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 

print ("The original array is: ", lst) 
print() 
numOfItems = 4
# remove first elements
result = lst[numOfItems:]
print ("The array after removing the elements is: ", result) 

输出

The original array is:  [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
The array after removing the elements is:  [5, 6, 7, 8, 9, 10]

从给定数组中移除前4个元素,并将结果数组存储在result变量中。在这个示例中,原始数组保持不变。

示例

通过使用Python的del关键字和切片对象,我们可以移除数组的元素。

# creating array
lst = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 
print ("The original array is: ", lst) 
print() 
numOfItems = 4

# remove first elements
del lst[:numOfItems]
print ("The array after removing the elements is: ", lst) 

输出

The original array is:  [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
The array after removing the elements is:  [5, 6, 7, 8, 9, 10]

语句lst[:numOfItems]检索数组中前几个给定数量的元素,而del关键字则移除这些元素。

使用NumPy数组

使用numpy模块和切片技术,我们可以轻松地从数组中移除几个元素。

示例

在这个示例中,我们将从NumPy数组中移除第一个元素。

import numpy
# creating array
numpy_array = numpy.array([1, 3, 5, 6, 2, 9, 8])
print ("The original array is: ", numpy_array) 
print() 
numOfItems = 3

# remove first elements
result = numpy_array[numOfItems:]
print ("The result is: ", result) 

输出

The original array is:  [1 3 5 6 2 9 8]

The result is:  [6 2 9 8]

我们已经成功地使用数组切片从NumPy数组中移除了前两个元素。

使用array模块

Python中的array模块也支持索引和切片技术来访问元素。

示例

在这个示例中,我们将使用array模块创建一个数组。

import array
# creating array
arr = array.array('i', [2, 1, 4, 3, 6, 5, 8, 7])
print ("The original array is: ", arr) 
print() 

numOfItems = 2
# remove first elements
result = arr[numOfItems:]
print ("The result is: ", result) 

输出

The original array is:  array('i', [2, 1, 4, 3, 6, 5, 8, 7])
The result is:  array('i', [4, 3, 6, 5, 8, 7])

结果数组已从数组arr中移除了前两个元素,这里数组arr保持不变。

更新于: 2023年5月29日

106 次浏览

开启你的职业生涯

通过完成课程获得认证

开始学习
广告