Python 是一种多功能且功能强大的编程语言,它提供了许多功能和语法,使编码更容易、更具表现力。但是,Python 中有一些运算符不是很常用或被许多程序员知道。
什么是海象运算符(warlus)
walrus 运算符,也称为赋值表达式运算符,是 Python 3.8 中引入的一项新功能。它允许为变量赋值并在同一个表达式中使用它,而无需编写两次。walrus 运算符的语法是:=
它可以在不允许使用常规赋值语句的地方使用,例如在 lambda 函数、列表推导式或条件表达式中。假设要编写一个函数来检查字符串是否为回文,即向前和向后读取时它是相同的。执行此操作的常见方法是反转字符串并将其与原始字符串进行比较,如下所示:
def is_palindrome(s):
return s == s[::-1]
但是,此方法会创建字符串的副本,这对于大型字符串可能效率低下。使用 walrus 运算符,可以通过将反转的字符串分配给变量并在同一表达式中使用它来避免这种情况,如下所示:
def is_palindrome(s):
return (rev := s[::-1]) == s
这样,只需要反转字符串一次,如果需要,可以将该变量用于其他目的。rev
为什么要使用海象运算符?
在希望避免重复计算或函数调用并保存一些代码行的情况下,walrus 运算符非常有用。例如,假设要编写一个函数来计算字符串中的元音数。执行此操作的常见方法是使用循环和计数器,如下所示:
def count_vowels(s):
vowels = "aeiou"
count = 0
for c in s:
if c in vowels:
count += 1
return count
但是,使用 walrus 运算符,可以通过使用列表推导式和 sum 函数来简化此操作,如下所示:
def count_vowels(s):
vowels = "aeiou"
return sum(c in vowels for c in s)
这样,只需要一行代码,并且不需要声明计数器变量。
海象运算符可以派上用场的另一种情况是,当想要使用依赖于条件的值,并避免重复计算条件。假设要编写一个函数,该函数返回两个数字的最大值,但如果它们相等,则返回它们的总和。执行此操作的常见方法是使用 if-else 语句,如下所示:
def max_or_sum(a, b):
if a == b:
return a + b
else:
return max(a, b)
但是,使用 walrus 运算符,您可以通过使用条件表达式来简化此操作,如下所示:
def max_or_sum(a, b):
return (a + b) if (equal := a == b) else max(a, b)
这样,只需要检查一次数字是否相等,如果需要,可以将该变量用于其他目的。equal
如何使用海象运算符?
walrus 运算符可用于任何允许变量名称的表达式,例如 lambda 函数、列表推导式、生成器表达式、集合推导式、字典推导式、条件表达式或函数调用。但是,在使用 walrus 运算符时,您需要注意一些规则和限制:
- 使用 walrus 运算符分配的变量名称必须是有效的 Python 标识符,即它必须以字母或下划线开头,并且只能包含字母、数字和下划线。不能使用关键字、文本或运算符作为变量名称。
- 使用 walrus 运算符分配的变量名称必须括在括号中,除非它是括起来表达式中的唯一表达式。例如,您可以写 ,但不能写 。(x := 1) + 2x := 1 + 2
- 使用 walrus 运算符分配的变量名称的范围限定为最近的封闭块,即可以在同一块内访问它,但不能在块外访问。例如,您可以写 ,但不能写 。if (x := 1) > 0: print(x)if x := 1: print(x)
- 不能在同一表达式中重新分配使用 walrus 运算符分配的变量名称,即不能在同一表达式中多次将同一变量名称与 walrus 运算符一起使用。例如,您可以写 ,但不能写 。(x := 1) + (y := 2)(x := 1) + (x := 2)
- 不能在同一表达式中删除使用 walrus 运算符分配的变量名称,即不能在同一表达式中使用使用 walrus 运算符分配的变量名称上的语句。例如,您可以写 ,但不能写 。del(x := 1) + 2del (x := 1)