python enumerate 函数

逃离我推掉我的手 2023-05-29 09:19 45阅读 0赞

## enumerate ##

这是一个有意思的内置函数，本来我们可以通过 `for i in range(len(list))`的方式得到一个 list 的每个元素索引，然后在用 list\[i\]的方式得到该元素。如果要同时得到元素索引和元素怎么办？就是这样了:

>>> for i in range(len(week)):
    ...     print week[i]+' is '+str(i)     #注意，i 是 int 类型，如果和前面的用 + 连接，必须是 str 类型
    ... 
    monday is 0
    sunday is 1
    friday is 2

Python 中提供了一个内置函数 enumerate，能够实现类似的功能

>>> for (i,day) in enumerate(week):
    ...     print day+' is '+str(i)
    ... 
    monday is 0
    sunday is 1
    friday is 2

官方文档是这么说的：

> Return an enumerate object. sequence must be a sequence, an iterator, or some other object which supports iteration. The next() method of the iterator returned by enumerate() returns a tuple containing a count (from start which defaults to 0) and the values obtained from iterating over sequence:

顺便抄录几个例子，供看官欣赏，最好实验一下。

>>> seasons = ['Spring', 'Summer', 'Fall', 'Winter']
    >>> list(enumerate(seasons))
    [(0, 'Spring'), (1, 'Summer'), (2, 'Fall'), (3, 'Winter')]
    >>> list(enumerate(seasons, start=1))
    [(1, 'Spring'), (2, 'Summer'), (3, 'Fall'), (4, 'Winter')]

对于这样一个列表：

>>> mylist = ["qiwsir",703,"python"]
    >>> enumerate(mylist)
    <enumerate object at 0xb74a63c4>

出现这个结果，用 list 就能实现转换，显示内容.意味着可迭代。

>>> list(enumerate(mylist))
    [(0, 'qiwsir'), (1, 703), (2, 'python')]

再设计一个小问题，练习一下这个函数。

**问题：**将字符串中的某些字符替换为其它的字符串。原始字符串"Do you love Canglaoshi? Canglaoshi is a good teacher."，请将"Canglaoshi"替换为"PHP".

**解析：**

>>> raw = "Do you love Canglaoshi? Canglaoshi is a good teacher."

这是所要求的那个字符串，当时，不能直接对这个字符串使用 `enumerate()`，因为它会变成这样：

>>> list(enumerate(raw))
    [(0, 'D'), (1, 'o'), (2, ' '), (3, 'y'), (4, 'o'), (5, 'u'), (6, ' '), (7, 'l'), (8, 'o'), (9, 'v'), (10, 'e'), (11, ' '), (12, 'C'), (13, 'a'), (14, 'n'), (15, 'g'), (16, 'l'), (17, 'a'), (18, 'o'), (19, 's'), (20, 'h'), (21, 'i'), (22, '?'), (23, ' '), (24, 'C'), (25, 'a'), (26, 'n'), (27, 'g'), (28, 'l'), (29, 'a'), (30, 'o'), (31, 's'), (32, 'h'), (33, 'i'), (34, ' '), (35, 'i'), (36, 's'), (37, ' '), (38, 'a'), (39, ' '), (40, 'g'), (41, 'o'), (42, 'o'), (43, 'd'), (44, ' '), (45, 't'), (46, 'e'), (47, 'a'), (48, 'c'), (49, 'h'), (50, 'e'), (51, 'r'), (52, '.')]

这不是所需要的。所以，先把 raw 转化为列表：

>>> raw_lst = raw.split(" ")

然后用 `enumerate()`

>>> for i, string in enumerate(raw_lst):
    ...     if string == "Canglaoshi":
    ...         raw_lst[i] = "PHP"
    ...

没有什么异常现象，查看一下那个 raw\_lst 列表，看看是不是把"Canglaoshi"替换为"PHP"了。

>>> raw_lst
    ['Do', 'you', 'love', 'Canglaoshi?', 'PHP', 'is', 'a', 'good', 'teacher.']

只替换了一个，还有一个没有替换。为什么？仔细观察发现，没有替换的那个是'Canglaoshi?',跟条件判断中的"Canglaoshi"不一样。

修改一下，把条件放宽：

>>> for i, string in enumerate(raw_lst):
    ...     if "Canglaoshi" in string:
    ...         raw_lst[i] = "PHP"
    ... 
    >>> raw_lst
    ['Do', 'you', 'love', 'PHP', 'PHP', 'is', 'a', 'good', 'teacher.']

好的。然后呢？再转化为字符串？留给读者试试。