5个 Python 切片技巧,让你的代码更加优雅
在Python世界中,有许多小技巧可以对我们的代码产生很大的影响。特别是对于大型程序,这些小技巧可以使我们的代码更整洁和优雅。
切片操作(slice operator)就是其中的一个。它可以帮助我们优雅的从可迭代对象中获取想要的元素,例如我们有一个如下的列表:
a = [1,2,3,4,5,6]
如果我们想获取列表 a 的奇数项该怎么写?
正常来说可以编写一个包含if-else语句的for循环语句过滤列表的元素,这个可行,就是多了许多不必要的代码。
其实这个问题只需要一行代码就可以解决:
a[::2]
没有 for 循环,没有 if-else 语句,只需一行。这就是切片技术。
下面就将介绍5个有用的Python切片技巧。
切片运算符的模板
在探索其用法之前,我们需要了解编写切片(slice)运算符的模板:
a_list[start:end:step]
如上所示,该操作包含三个参数:
start: 起始索引 (默认值是 0)
end: 结束索引 (默认值是该列表的长度)
step: 步长 (默认值是 1)
这三个参数都是可以选项。
了解了模板之后,现在让我们看看这5个技巧是如何工作的。
1.获取列表的前N个或后N个项目
我们再次使用上一个列表作 a 为示例:
a = [1,2,3,4,5,6]
如何获取前三个项?很简单:
a[:3] # 输出: [1, 2, 3]
在上面的代码中,start 从 0 开始,也就是该参数的默认值,所以我们就写 end 索引是多少
那如何获取后三个项目呢? 这里就用到了负数索引:
a[-3:] # 输出: [4, 5, 6]
众所周知,大多数编程语言都支持正索引,并且索引从0开始。
但是 Python 是个很特别的语言,它支持负索引。它从 -1 开始,换句话说,-1代表最后一个项目,-2代表倒数第二个项目,依此类推。
因此,当我们需要获取列表的最后n个项目时,只需通过负索引即可获取它。
2.取列表的第N个项目
我们可以通过第三个参数步长(step), 更加灵活的处理切片任务,比如获取列表的奇数项:
a = [1, 2, 3, 4, 5, 6]a[::2] # 输出 [1, 3, 5]
这里有个常用的小技巧就是通过它翻转列表:
a = [1, 2, 3, 4, 5, 6]a[::-1] # 输出 [6, 5, 4, 3, 2, 1]
正步数是从列表头开始,负步数就是从列表最后开始。
3.浅复制列表
浅表复制的最简单方法是使用以下切片技巧:
a = [1, 2, 3, 4, 5, 6]b = a[:]b[0]=100print(b) # 输出 [100, 2, 3, 4, 5, 6]print(a) # 输出 [1, 2, 3, 4, 5, 6]
4. 一次操作多个列表元素
如果我们需要处理列表中的多个元素,是否必须使用 for 循环?
并不一定,我们仍然可以通过切片运算符执行某些任务,例如分配或删除而无需for循环。我们来看一些例子:
一次删除多个列表元素
a = [1, 2, 3, 4, 5, 6]del a[:2]print(a) # 输出 [3, 4, 5, 6]
调整列表大小
a = [1, 2, 3, 4, 5, 6]a[:4] = [7,8]print(a) # 输出 [7, 8, 5, 6]
5.使用切片对象(Slice Object)存储切片索引
如果我们用相同的索引对许多列表进行切片,那么可能会有太多的数字,即使借助切片技巧,我们的代码也会变得难看。像下面的代码
a = [1, 2, 3, 4, 5, 6]b = [3, 4, 5, 6, 7, 8, 9]c = [2, 3, 4, 5, 100, 101, 102]a2 = a[1:5:2]b2 = b[1:5:2]c2 = c[1:5:2]
更不用说如果所需的索引更改了,我们必须一一修改相应的索引。
一般来说,在一个程序中放入许多相同的数字不是一个好习惯。 这种情况下就可以使用切片对象(Slice Object)来存储切片索引。
a = [1, 2, 3, 4, 5, 6]b = [3, 4, 5, 6, 7, 8, 9]c = [2, 3, 4, 5, 100, 101, 102]indexes = slice(1, 5, 2)a2 = a[indexes]b2 = b[indexes]c2 = c[indexes]
如上所示,内置的切片对象是一个简单的辅助工具,可以帮助我们存储三个数字并在需要时使用它。如果索引更改,我们只需要修改slice对象中的数字即可。
总结
Python 为我们提供了一些语法糖,可以使我们的代码更具可读性和优雅性。切片技术就是一个很好的例子。了解并应用这些小技巧可以极大地提高我们的代码质量。