Python 是一种非常容易学习的语言,它不会强迫您使用特定的约定。也很容易陷入一些陷阱,使你作为一个初学者脱颖而出。为避免看起来像一个完全的初学者,请查看下面的提示以及如何修复它们。
1.不使用 enumerate()
Python 使得遍历列表之类的东西变得超级容易。请看下面的例子:
vals = ['Python', 'is', 'fun!']
for val in vals:
print(val)
# Returns:
# Python
# is
# fun!
但是,如果您需要引用每个项目的索引,会发生什么情况?您可能已经遇到甚至编写了如下所示的代码:
vals = ['Python', 'is', 'fun!']
for i in range(len(vals)):
print(i, vals[i])
# Returns:
# 0 Python
# 1 is
# 2 fun!
虽然这可行,但它并不完全优雅!它也不会立即清楚地表明您的代码正在做什么。
这就是 'enumerate()' 函数派上用场的地方!让我们来看看它使这个过程变得多么容易:
vals = ['Python', 'is', 'fun!']
for idx, val in enumerate(vals):
print(idx, val)
# Returns:
# 0 Python
# 1 is
# 2 fun!
那不是更好吗?
作为专业提示,您甚至可以更改起始值。假设您想改为从 1 开始标记项目。然后你可以简单地写:
vals = ['Python', 'is', 'fun!']
for idx, val in enumerate(vals, start=1):
print(idx, val)
# Returns:
# 1 Python
# 2 is
# 3 fun!
好了,让我们深入了解下一个初学者习惯吧!
2.不使用三元算子
当你是一个 if-else 语句来赋值时,你就使用了大量的代码行来执行一个简单的操作。请看一下下面的场景:
amount = 100
if amount > 50:
raise_amount = 20
else:
raise_amount = 10
print(raise_amount)
# Returns:
# 20
在上面的示例中,我们正在创建一个 if-else 语句,用于检查某人的销售金额是否超过 50。如果是这样,那么他们会得到 20 的加薪。否则,他们将获得 10 的加薪。
虽然我们的代码很干净,但它并不完全简洁。我们可以通过使用三元赋值来大大简化这一点:
amount = 100
raise_amount = 20 if amount > 50 else 10
print(raise_amount)
# Returns:
# 20
这最适合非常简单的任务。虽然你可以变得更复杂,但不要牺牲可读性而不是简洁性!
3.不使用推导式
Python 推导式可以缩短创建列表、字典甚至生成器的工作。它们提供了一种优雅且可读的方式来轻松创建数据结构。
让我们看一个示例,说明如何使用 for 循环来编写创建平方数列表:
squares = []
for i in range(1, 6):
squares.append(i ** 2)
print(squares)
# Returns:
# [1, 4, 9, 16, 25]
现在,让我们将其与列表推导进行比较:
squares = [i ** 2 for i in range(1, 6)]
print(squares)
# Returns:
# [1, 4, 9, 16, 25]
我们可以看到使用列表推导是多么简单和明确!不仅代码更简洁,而且更易于阅读。
创建字典理解同样容易。让我们通过创建一个字典来再次比较这两种方法,其中键是原始数字,值是它的平方:
squares = {}
for i in range(1, 6):
squares[i] = i ** 2
print(squares)
# Returns:
# {1: 1, 2: 4, 3: 9, 4: 16, 5: 25}
现在让我们来看看如何通过字典理解来简化这一点!
squares = {i: i**2 for i in range(1, 6)}
print(squares)
# Returns:
# {1: 1, 2: 4, 3: 9, 4: 16, 5: 25}
现在,也很容易过火,对所有事情都使用理解。因此,我在这里提出一个额外的提示:除非有意义,否则不要将它们用于所有事情!
3b. 过度理解
Python 推导式很强大——真的很强大。您可以构建复杂的 if-else 语句,甚至可以将它们相互嵌套。
让我们看一下如何合并 if-else 语句:
even_odd_numbers = [
"Even" if num % 2 == 0 else "Odd"
for num in range(10)
]
print(even_odd_numbers)
# Returns:
# ['Even', 'Odd', 'Even', 'Odd', 'Even', 'Odd', 'Even', 'Odd', 'Even', 'Odd']
在上面的示例中,我们使用三元运算符在继续迭代的同时返回一个值。现在,这个例子非常简单,但让我们看一个更复杂的例子:
result = [x * 2 if x % 2 == 0 else x * 3 for x in range(10) if x != 5]
哎呀!好吧,读起来一点也不好玩。这就引出了我的主要观点——如果它们不如使用 for 循环清晰,就不要编写推导式。
让我们将这种理解转换为一个循环,看看阅读起来有多容易:
result = []
for x in range(10):
if x != 5:
if x % 2 == 0:
result.append(x * 2)
else:
result.append(x * 3)
我们可以看到,虽然我们使用更多的行来编写此代码,但它的可读性大大提高!
4. 不使用 itertools
Python 的 itertools 是 Python 中内置的隐藏宝石。虽然从表面上看,它的许多功能看起来很简单,但它们提供了优雅而强大的方法来迭代不同的对象。
防止嵌套循环
假设您有两个列表,并且您想要遍历所有可能的组合。您可以编写一个嵌套的 for 循环,如下所示:
colors = ['Red', 'Green']
sizes = ['S', 'M', 'L']
for color in colors:
for size in sizes:
print(color, size)
# Returns:
# Red S
# Red M
# Red L
# Green S
# Green M
# Green L
与我们在本文中介绍的其他所有内容类似,这种方法是有效的,但它并不完全优雅。
值得庆幸的是,itertools 带有 product() 函数,该函数创建所有项的笛卡尔乘积。这意味着我们可以直接在单个 for 循环中解压缩值,如下所示:
from itertools import product
colors = ['Red', 'Green']
sizes = ['S', 'M', 'L']
for color, size in product(colors, sizes):
print(color, size)
# Returns:
# Red S
# Red M
# Red L
# Green S
# Green M
# Green L
我们可以看到这种方法要简单得多!
成对循环访问列表
在某些情况下,您需要成对地迭代列表,这意味着您需要访问一个项目及其邻居。
为此,您可以编写以下代码:
vals = [1, 2, 3, 4]
for i in range(len(vals) - 1):
print((vals[i], vals[i+1]))
# Returns:
# (1, 2)
# (2, 3)
# (3, 4)
这种方法效果很好,但它不容易阅读,而且它的作用也不是很明确。
这就是 Python 3.10 中引入的 'pairwise()' 函数派上用场的地方!让我们看一下如何简化我们的代码:
from itertools import pairwise
vals = [1, 2, 3, 4]
for pair in pairwise(vals):
print(pair)
# Returns:
# (1, 2)
# (2, 3)
# (3, 4)
这是我们之前编写的代码的更明确的版本,使我们能够立即理解代码的用途。
itertools 库为您提供了许多用于迭代对象的有用函数。知道何时应用这些功能确实将您的技能提升到另一个层次。
结论
掌握 Python 不仅仅是记住语法,而是拥抱优雅,并知道何时在简洁性和可读性之间取得平衡。