拨开荷叶行,寻梦已然成。仙女莲花里,翩翩白鹭情。
IMG-LOGO
主页 文章列表 10个小技巧,让你的 Python 代码更加优雅

10个小技巧,让你的 Python 代码更加优雅

白鹭 - 2022-02-28 2076 0 0

10个小技巧改进的 Python 代码,让你的代码更加简洁、更加 Python 化,

1. 用enumerate代替range

如果你需要遍历一个串列,并且需要同时获取索引和元素,大多数情况可能会使用 range(len(data)) 的语法,

比如我们要遍历一个串列,找到所有的负数:

很多人学习蟒蛇,不知道从何学起,

很多人学习python,掌握了基本语法之后,不知道在哪里寻找案例上手,

很多已经可能案例的人,却不知道如何去学习更多高深的知识,

那么针对这三类人,我给大家提供一个好的学习平台,免费获取视频教程,电子书,以及课程的源代码!

QQ群:101677771

欢迎加入,一起讨论一起学习

 

data = [1, 3, -5, 7, 9, -11]
for i in range(len(data)):
	if data[i] < 0:
		print(f"索引:{
   i},元素:{
   data[i]}")

输出:

索引:2,元素:-5
索引:5,元素:-11

虽然这种方法有效,但如果使用 Python 内置的 enumerate 函式会更好,这个函式会回传遍历程序中当前索引和当前元素值作为一个元组,因此可以直接输出索引和值:

data = [1, 3, -5, 7, 9, -11]
for i, v, in enumerate(data):
	if v < 0:
		print(f"索引:{
   i},元素:{
   v}")

输出:

索引:2,元素:-5
索引:5,元素:-11

2. 深入理解list,替代for回圈

假如我们要创建一个从0~9的平方数的串列,一种简单的方法是先创建一个 list ,然后通过 for 回圈每次将一个数的平方 append 到串列的结尾,

squares = []
for i in range(10):
	squares.append(i ** 2)
print(squares)

输出:

[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

但其实通过串列生成式,我们有一种更快捷的方法,只需要一行代码就能够实作相同的功能,

squares = [i ** 2 for i in range(10)]
print(squares)

输出:

[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

3. 通过 Python 内置的sorted()方法对复杂的物件进行排序

排序的需求在很多场景中都很常见,Python中的可迭代物件(串列、元组、字典)都可以进行排序,内置函式 sorted() 让我们甚至都不需要自己实作排序算法,

data = [1, 3, -5, 7, 9, -11]
sortedData = sorted(data)
print(sortedData)

输出:

[-11, -5, 1, 3, 7, 9]

sorted() 方法自动将串列按照升序排序,如果你想按照降序排序,可以使用 sorted() 的自变量 reverse = True ,

sorted() 方法同样也适用于元组型别的资料,但是注意排序的结果将回传一个串列,

data = (1, 3, -5, 7, 9, -11)
print(data)
sortedData = sorted(data)
print(sortedData)

输出:

(1, 3, -5, 7, 9, -11)
[-11, -5, 1, 3, 7, 9]

对于复杂的可迭代物件,比如我们可以创建一个串列,串列中的每一个元素都是一个人的字典信息,然后我们按照每个人的年龄进行排序,

我们可以用 sorted() 的 key 自变量,传入一个匿名函式,让排序按照我们预想的方式进行,

data = https://www.cnblogs.com/sn520/archive/2022/01/04/[{
 "name": "Alex", "age": 18},
        {
 "name": "Band", "age": 21},
        {
 "name": "Coco", "age": 17}]
sorted_data = https://www.cnblogs.com/sn520/archive/2022/01/04/sorted(data, key=lambda x: x["age"])
print(sorted_data)

输出:

[{'name': 'Coco', 'age': 17}, {'name': 'Alex', 'age': 18}, {'name': 'Band', 'age': 21}]

4. 使用集合Set存盘唯一值

如果你有一个包含多个值的串列,并且需要去重,一个不错的技巧是将我们的串列转换为一个集合,

Python的集合是一种无序的资料型别,并且没有重复元素,还提供了不同集合之间的运算,可以求交集、并集和差集,

data = [1, 3, 3, 5, 5, 5, 7, 7, 7, 7, 9, 9, 9, 9, 9]
setData = set(data)
print(setData)

输出:

{1, 3, 5, 7, 9}

5. 使用生成器以节省存储器

有时候串列并不是最好的选择,假如我们有一个非常大的串列,有10000个元素,我们想计算所有元素的总和,虽然可以用串列来做,当如果资料量特别大的话我们可能会遇到存储器问题,更好的解决方案是使用生成器,

dataList = [i for i in range(10000)]
print(sum(dataList))
dataGen = (i for i in range(10000))
print(sum(dataGen))

输出:

串列生成式与生成器具有相同的语法,不同点在于串列生成式使用的是中括号,而生成器使用的是小括号,

生成器通过类似于懒加载的方式计算我们需要的元素,因此它一次只生成一个元素,并且只在需要的时候才生成,

我们可以使用 sys 的 getsize() 方法看一下串列和生成器的大小,

import sys

dataList = [i for i in range(10000)]
print(sys.getsizeof(dataList), "bytes")

dataGen = (i for i in range(10000))
print(sys.getsizeof(dataGen), "bytes")

输出:

85176 bytes
112 bytes

6. 在字典中获取元素使用.get()并且设定默认值

假设我们有一个字典,其中包括不同的键,比如商品和价格,在我们的代码中的某个点,想要获得商品的价格,

当我们用 [] 简单的访问键时,如果键不在字典中,那么代码将会报错,引发一个 KeyError ,

dataDict = {
 
        "name": "Tesla",
        "price": 250000
}
print(dataDict["count"])

报错:

Traceback (most recent call last):
  File "mian.py", line 5, in <module>
    print(dataDict["count"])
KeyError: 'count'

更好的方式是使用 .get() 方法,此时如果键不存在,则不会引发 KeyError ,而是回传一个默认值,如果我们不指定默认值,它会直接回传None,如果我们指定默认值,那么它会回传我们指定的默认值,

dataDict = {
 
        "name": "Tesla",
        "price": 250000
}
print(dataDict.get("count"))
print(dataDict.get("count", 0))

输出:

None
0

7. 计数器collections.Counter

如果你想计算串列中元素的数量,Python中有一个非常方便的工具,

from collections import Counter

data = [1, 3, 3, 5, 5, 5, 7, 7, 7, 7, 9, 9, 9, 9, 9]
counter = Counter(data)

print(counter)

输出:

Counter({9: 5, 7: 4, 5: 3, 3: 2, 1: 1})

Counter 可以统计串列中不同元素的数量,并且按照出现次数降序排列,这比我们自己计算要好得多,

如果你想知道某个元素的个数的话,可以直接通过 [] 访问,如果这个元素不存在,那么将回传0,并且还可以通过 most_common() 方法回传排名靠前的几个元素,

from collections import Counter

data = [1, 3, 3, 5, 5, 5, 7, 7, 7, 7, 9, 9, 9, 9, 9]
counter = Counter(data)

print(counter[7])
print(counter[11])
print(counter.most_common(2))

输出:

4
0
[(9, 5), (7, 4)]

8. 使用f-String格式化字符串(适用于Python 3.6+)

f-String在我看来是最佳的格式化字符串方法,我们只需要在字符串前写一个f,然后在字符串内部可以直接使用大括号来嵌入变量或表达式,

data = https://www.cnblogs.com/sn520/archive/2022/01/04/{
 "name": "Alex", "age": 18}
string = f"I'm {
   data['name']} and I am {
   data['age']} years old."
print(string)

输出:

I'm Alex and I am 18 years old.

这种方式更简单、更简洁、速度也更快,

9. 使用.join()拼接字符串

假如我们有一个包含不同字符串的串列,我们想把所有的字符串通过空格拼接在一起,千万不要再用 for 回圈一个一个遍历然后拼接了,一种更简洁的方法是通过 .joint() 方法,它可以使用指定的字符将字符串拼接起来,

strings = ["Hello", "World", "!"]
print(" ".join(strings))
print("_".join(strings))

输出:

Hello World !
Hello_World_!

10. 使用双星号语法合并字典(适用于Python 3.5+)

如果我们有两个字典,并且希望将其合并成一个,此时我们可以使用双星号 ** 和大括号 {} 语法,

info1 = {
 "name": "Alex", "age": 18}
info2 = {
 "name": "Alex", "city": "Bei Jing"}
info = {
 **info1, **info2}
print(info)

输出:

{'name': 'Alex', 'age': 18, 'city': 'Bei Jing'}

以上就是这篇文章跟大家分享的小技巧,

如果你觉得这篇文章对你有帮助的话,欢迎一键三连~

标签:

0 评论

发表评论

您的电子邮件地址不会被公开。 必填的字段已做标记 *