跳至主要內容

4. 流程控制

Zhao Bin笔记backendpython

4. 流程控制

4.1 if 语句

if 语句包含零个或多个 elif 子句及可选的 else 子句。

x = int(input('Please enter an integer: '))
if x < 0:
  x = 0
  print('Negative changed to zero')
elif x == 0:
  print('Zero')
elif x == 1:
  print('Single')
else:
  print('More')

如果要把一个值与多个常量进行比较,或者检查特定类型或属性,match 语句更实用。

4.2 for 语句

Python 的 for 语句不迭代算术递增数值,或是给予用户定义迭代步骤和暂停条件的能力(如 C),而是迭代列表或字符串等任意序列,元素的迭代顺序与在序列中出现的顺序一致。

words = ['cat', 'window', 'test']
for w in words:
  print(w, len(w))

遍历集合时修改集合的内容,会很容易生成错误的结果。因此不能直接进行循环,而是应遍历该集合的副本或创建新的集合:

users = {'Hans': 'active', 'hello': 'inactive', '赵斌': 'active'}

for user, status in users.copy().items():
  if status == 'inactive':
    del users[user]

active_users = {}
for user, status in users.items():
  if status == 'active':
    active_users[user] = status

print('\nactive users:')
for user in active_users:
  print(user)

试一试:

python 4.2/for.py

4.3 range 函数

内置函数 range() 常用于遍历数字序列。
range 可以不从 0 开始,还可以按指定幅度递增(递增幅度称为 '步进',支持负数)

for i in range(5):
  print(i)

sum(range(5))
# 0 + 1 + 2 + 3 + 4

list(range(5, 10))
# [5, 6, 7, 8, 9]

list(range(0, 10, 3))
# [0, 3, 6, 9]

list(range(-10, -100, -30))
# [-10, -40, -70]

range()len() 组合在一起,可以按索引迭代序列:

a = ['Mary', 'had', 'a', 'little', 'lamb']
for i in range(len(a)):
  print(i, a[i])

4.4 循环中的 break、continue 语句及 else 子句

break 语句和 C 中的类似,用于跳出最近的 for 或 while 循环。

循环语句支持 else 子句;for 循环中,可迭代对象中的元素全部循环完毕,或 while 循环的条件为假时,执行该子句;break 语句终止循环时,不执行该子句。

请看下面这个查找素数的循环示例:

for n in range(2, 10):
  for x in range(2, n):
    if n % x == 0:
      print(n, 'equals', x, '*', n // x)
      break
  else:
    # loop fell through without finding a factor
    print(n, 'is a prime number')

(没错,这段代码就是这么写。仔细看:else 子句属于 for 循环,不属于 if 语句。)

if 语句相比,循环的 else 子句更像 tryelse 子句: tryelse 子句在未触发异常时执行,循环的 else 子句则在未运行 break 时执行。try 语句和异常详见 异常的处理。

试一试:

python 4.4/else.py

4.5 pass

pass 语句不执行任何动作。语法上需要一个语句,但程序毋需执行任何动作时,可以使用该语句。

这常用于创建一个最小的类:

class MyEmptyClass:
  pass

pass 还可用作函数或条件语句体的占位符,让你保持在更抽象的层次进行思考。pass 会被默默地忽略:

def initlog(*args):
  pass  # Remember to implement this!

4.6 match 语句

match 语句接受一个表达式并把它的值与一个或多个 case 块给出的一系列模式进行比较。
这表面上像 C、Java 或 JavaScript(以及许多其他程序设计语言)中的 switch 语句,但其实它更像 Rust 或 Haskell 中的模式匹配。
只有第一个匹配的模式会被执行,并且它还可以提取值的组成部分(序列的元素或对象的属性)赋给变量。

最简单的形式是将一个主语值与一个或多个字面值进行比较:

def http_error(status):
  match status:
    case 400:
      return "Bad request"
    case 404:
      return "Not found"
    case 418:
      return "I'm a teapot"
    case 401 | 403: # 你可以使用 | (“ or ”)在一个模式中组合几个字面值:
      return "Not allowed"
    case _:
      return "Something's wrong with the internet"

注意最后一个代码块:“变量名” _ 被作为 通配符 并必定会匹配成功。如果没有 case 匹配成功,则不会执行任何分支。

试一试:

python 4.6/match-http-error.py 400

形如解包赋值的模式,可用于绑定变量:

# point is an (x, y) tuple
match point:
  case (0, 0):
    print("Origin")
  case (0, y):
    print(f"Y={y}")
  case (x, 0):
    print(f"X={x}")
  case (x, y):
    print(f"X={x}, Y={y}")
  case _:
    raise ValueError("Not a point")
from enum import Enum
class Color(Enum):
  RED = 'red'
  GREEN = 'green'
  BLUE = 'blue'

color = COLOR(input("Enter your choice of 'red', 'green' or 'blue': "))

match color:
  case COLOR.RED:
    print("I see red!")
  case COLOR.GREEN:
    print("Grass is green.")
  case COLOR.BLUE:
    print("I'm feeling the blues :(")

试一试:

python 4.6/match-point.py

4.7 定义函数

下列代码创建一个可以输出限定数值内的斐波那契数列函数:

def fib(n): # write Fibonacci series up to n
  """Print a Fibonacci series up to n."""
  a, b = 0, 1
  while a < n:
    print(a, end = " ")
    a,b = b, a + b
  print()

# Now call the function we just defined:
fib(10)

定义 函数使用关键字 def,后跟函数名与括号内的形参列表。函数语句从下一行开始,并且必须缩进。

函数内的第一条语句是字符串时,该字符串就是文档字符串,也称为 docstring,详见 文档字符串open in new window

fib 不返回值,因此,其他语言不把它当作函数,而是当作过程。事实上,没有 return 语句的函数也返回值,只不过这个值比较是 None (是一个内置名称)。一般来说,解释器不会输出单独的返回值 None ,如需查看该值,可以使用 print()

print(fib(0))
# None

4.8 函数详解

函数定义支持可变数量的参数。这里列出三种可以组合使用的形式。

4.8.1 默认参数

为参数指定默认值是非常有用的方式。调用函数时,可以使用比定义时更少的参数,例如:

def ask_ok(prompt, retries=4, reminder="Please try again!"):
  while True:
    ok = input(prompt)
    if ok in ("y", "ye", "yes"):
      return True
    if ok in ("N", "no", "nop", "nope"):
      return False
    retries = retries - 1
    if retries < 0:
      raise ValueError("invalid user response")
    print(reminder)

该函数可以用以下方式调用:

  • 只给出必选参数
    ask_ok('Do you really want to quit?')
  • 给出一个可选参数
    ask_ok('OK to overwrite the file?', 2)
  • 给出所有参数
    ask_ok('OK to overwrite the file?', 2, 'Come on, only yes or no!')

本例还使用了关键字 in ,用于确认序列中是否包含某个值。

注意

重要警告: 默认值只计算一次。默认值为列表、字典或类实例等可变对象时,会产生与该规则不同的结果。

例如,下面的函数会累积后续调用时传递的参数:

i = 5

def f(arg=i):
  print(arg)

i = 6
f()
# 输出: 5

def f(a, L=[]):
  L.append(a)
  return L

print(f(1))
print(f(2))
print(f(3))

# 输出结果:
# [1]
# [1, 2]
# [1, 2, 3]

不想在后续调用之间共享默认值时,应以如下方式编写函数:

def f(a, L=None):
  if L is None:
    L = []
  L.append(a)
  return L

4.8.2 关键字参数

kwarh=value 形式的关键字参数open in new window也可以用于调用参数。函数示例如下:

def parrot(voltage, state='a stiff', action='voom', type='Norwegian Blue'):
  print("-- This parrot wouldn't", action, end = ' ')
  print("if you put", voltage, "volts through it.")
  print("-- Lovely plumage, the", type)
  print("-- It's", state, "!")

该函数接受一个必选参数(voltage)和三个可选参数(state, action 和 type)。该函数可用下列方式调用:

parrot(1000)                                          # 1 positional argument (位置参数)
parrot(voltage=1000)                                  # 1 keyword argument (关键字参数)
parrot(voltage=1000000, action='VOOOOOM')             # 2 keyword arguments (关键字参数)
parrot(action='VOOOOOM', voltage=1000000)             # 2 keyword arguments (关键字参数)
parrot('a million', 'bereft of life', 'jump')         # 3 positional arguments (位置参数)
parrot('a thousand', state='pushing up the daisies')  # 1 positional, 1 keyword (位置参数,关键字参数)

以下调用函数的方式都无效:

parrot()                     # required argument missing
parrot(voltage=5.0, 'dead')  # non-keyword argument after a keyword argument
parrot(110, voltage=220)     # duplicate value for the same argument
parrot(actor='John Cleese')  # unknown keyword argument
  • 函数调用时,关键字参数必须跟在位置参数后面。
  • 所有传递的关键字参数都必须匹配一个函数接受的参数,关键字参数的顺序并不重要。

4.8.3 特殊参数

默认情况下,参数可以按位置或显式关键字传递给 Python 函数。为了让代码易读、高效,最好限制参数的传递方式,这样,开发者只需查看函数定义,即可确定参数项是仅按位置、按位置或关键字,还是仅按关键字传递。

函数定义如下:

def f(pos1, pos2, /, pos_or_kwd, *, kwd1, kwd2):
      -----------    ----------     ----------
        |             |                  |
        |        Positional or keyword   |
        |                                - Keyword only
         -- Positional only

/* 是可选的。这些符号表明形参如何把参数值传递给函数:位置、位置或关键字、关键字。关键字形参也叫作命名形参。

4.8.3.1 位置或关键字参数

函数定义中未使用 /* 时,参数可以按位置或关键字传递给函数。

4.8.3.2 仅位置参数

此处再介绍一些细节,特定形参可以标记为 仅限位置。仅限位置 时,形参的顺序很重要,且这些形参不能用关键字传递。仅限位置形参应放在 / (正斜杠)前。/ 用于在逻辑上分割仅限位置形参与其它形参。如果函数定义中没有 /,则表示没有仅限位置形参。

/ 后可以是 位置或关键字 或 仅限关键字 形参。

4.8.3.3 仅限关键字参数

把形参标记为 仅限关键字,表明必须以关键字参数形式传递该形参,应在参数列表中第一个 仅限关键字 形参前添加 *

4.8.4 任意实参列表

调用函数时,使用任意数量的实参是最少见的选项。这些实参包含在元组中(详见 元组和序列 )。在可变数量的实参之前,可能有若干个普通参数:

def write_multiple_items(file, separator, *args):
    file.write(separator.join(args))

variadic 参数用于采集传递给函数的所有剩余参数,因此,它们通常在形参列表的末尾。*args 形参后的任何形式参数只能是仅限关键字参数,即只能用作关键字参数,不能用作位置参数:

def concat(*args, sep="/"):
    return sep.join(args)

concat("earth", "mars", "venus")
# 'earth/mars/venus'
concat("earth", "mars", "venus", sep=".")
# 'earth.mars.venus'