迭代(遍历)
迭代字典
要遍历字典的所有关键字,可像遍历序列那样使用普通的for
语句。
d = {'x': 1, 'y': 2, 'z': 3}
for key in d:
print(key, 'corresponds to', d[key]
也可使用keys
等字典方法来获取所有的键。如果只对值感兴趣,可使用d.values
。你可能还记得,d.items
以元组的方式返回键-值对。for
循环的优点之一是,可在其中使用序列解包。
for key, value in d.items():
print(key, 'corresponds to', value)
注意 字典元素的排列顺序是不确定的。换而言之,迭代字典的键或值时,一定会处理所有的键或值,但不知道处理的顺序。如果顺序很重要,可将键或值存储在一个列表中并对列表排序,再进行迭代。要让映射记住其项的插入顺序,可使用模块collections
中的OrderedDict
类。
一些迭代工具
Python提供了多个可帮助迭代序列(或其他可迭代对象)的函数。更具体的在模块itertools中。
并行迭代
有时候,你可能想同时迭代两个序列。假设有下面两个列表:
names = ['anne', 'beth', 'george', 'damon'] ages = [12, 45, 32, 102]
如果要打印名字和对应的年龄,可以像下面这样做:
for i in range(len(names)): print(names[i], 'is', ages[i], 'years old')
i是用作循环索引的变量的标准名称。
一个很有用的并行迭代工具是内置函数zip
,它将两个序列“缝合”起来,并返回一个由元组组成的序列。返回值是一个适合迭代的对象,要查看其内容,可使用list
将其转换为列表。
>>> list(zip(names, ages)) [('anne', 12), ('beth', 45), ('george', 32), ('damon', 102)]
“缝合”后,可在循环中将元组解包。
for name, age in zip(names, ages): print(name, 'is', age, 'years old')
函数zip
可用于“缝合”任意数量的序列。需要指出的是,当序列的长度不同时,函数zip
将在最短的序列用完后停止“缝合”。
>>> list(zip(range(5), range(100000000)))
[(0, 0), (1, 1), (2, 2), (3, 3), (4, 4)]
迭代时获取索引
在有些情况下,你需要在迭代对象序列的同时获取当前对象的索引。例如,你可能想替换一个字符串列表中所有包含子串'xxx'
的字符串。当然,完成这种任务的方法有很多,但这里假设你要像下面这样做:
for string in strings: if 'xxx' in string: index = strings.index(string) # 在字符串列表中查找字符串 strings[index] = '[censored]'
这可行,但替换前的搜索好像没有必要。另外,如果没有替换,搜索返回的索引可能不对(即返回的是该字符串首次出现处的索引)。下面是一种更佳的解决方案:
index = 0 for string in strings: if 'xxx' in string: strings[index] = '[censored]' index += 1
这个解决方案虽然可以接受,但看起来也有点笨拙。另一种解决方案是使用内置函数enumerate
。
for index, string in enumerate(strings): if 'xxx' in string: strings[index] = '[censored]'
这个函数让你能够迭代索引-值对,其中的索引是自动提供的。
反向迭代和排序后再迭代
来看另外两个很有用的函数:reversed
和sorted
。它们类似于列表方法reverse
和sort
(sorted
接受的参数也与sort
类似),但可用于任何序列或可迭代的对象,且不就地修改对象,而是返回反转和排序后的版本。
>>> sorted([4, 3, 6, 8, 3]) [3, 3, 4, 6, 8] >>> sorted('Hello, world!') [' ', '!', ',', 'H', 'd', 'e', 'l', 'l', 'l', 'o', 'o', 'r', 'w'] >>> list(reversed('Hello, world!')) ['!', 'd', 'l', 'r', 'o', 'w', ' ', ',', 'o', 'l', 'l', 'e', 'H'] >>> ''.join(reversed('Hello, world!')) '!dlrow ,olleH'
请注意,sorted
返回一个列表,而reversed
像zip
那样返回一个更神秘的可迭代对象。你无需关心这到底意味着什么,只管在for
循环或join
等方法中使用它,不会有任何问题。只是你不能对它执行索引或切片操作,也不能直接对它调用列表的方法。要执行这些操作,可先使用list
对返回的对象进行转换。
提示 要按字母表排序,可先转换为小写。为此,可将
sort
或sorted
的key
参数设置为str.lower
。例如,sorted("aBc", key=str.lower)
返回['a', 'B', 'c']
。
使用break直接跳出循环
continue
语句continue
没有break
用得多。它结束当前迭代,并跳到下一次迭代开头。这基本上意味着跳过循环体中余下的语句,但不结束循环。这在循环体庞大而复杂,且存在多个要跳过它的原因时很有用。在这种情况下,可使用continue.