可迭代对象和迭代器

判断

可通过模块collections.abc和isinstance函数来验证一个对象是否是可迭代(Iterable)、迭代器(Iterable)以及生成器(Generator)。

可迭代对象

常见的对象比如string、list、tuple、dict都是Iterable对象,他们都可以for循环遍历每一个元素:

from collections.abc import Iterable, Iterator, Generator
obj_1 = 'abcde' # string
obj_2 = [1,2,3,4,5] # list
obj_3 = (5,4,3,2,1) #tuple
obj_4 = {'1':1, '2':2} #dict
obj_5 = 4 #int
print('string %s Iterable' % ('is' if isinstance(obj_1, Iterable) else 'is not'))
print('list %s Iterable' % ('is' if isinstance(obj_2, Iterable) else 'is not'))
print('tuple %s Iterable' % ('is' if isinstance(obj_3, Iterable) else 'is not'))
print('dict %s Iterable' % ('is' if isinstance(obj_4, Iterable) else 'is not'))
print('int %s Iterable' % ('is' if isinstance(obj_5, Iterable) else 'is not'))

输出结果为:

string is Iterable
list is Iterable
tuple is Iterable
dict is Iterable
int is not Iterable

Iterable对象的本质是该对象内部实现了一个__iter__方法,也可使用dir函数通过判断该对象是否有__iter__方法来判别是否为Iterable对象。使用for循环实际上干了两件事,首先调用__iter__返回一个对象,该对象必须有__next__方法,然后再循环调用该对象的__next__方法不断获取下一个值。

迭代器

同时实现了__iter__、__next__方法的对象为迭代器,若一个对象定义了__next__方法,则除了for循环外,还可以将该对象作为next方法的输入参数以获取下一个迭代值。
下面的例子定义了一个可迭代对象NaturalNumberList,他接受一个自然数,然后存储了到该自然数的所有自然数。

from collections.abc import Iterable, Iterator, Generator
class NaturalNumberList:
        def __init__(self, num):
                self.num = num
        def __iter__(self):
                return NaturalNumberIterator(self.num) # 返回一个跌代器
class NaturalNumberIterator:
        def __init__(self, num):
                self.num = num
                self.currentValue = -1
        def __iter__(self):
                return self
        def __next__(self):
                if self.currentValue < self.num:
                        self.currentValue += 1
                        return self.currentValue # 每次调用返回该次循环的值
                else:
                        raise StopIteration() # 循环结束
naturalNumberList = NaturalNumberList(5)
naturalNumberIterator = NaturalNumberIterator(5)
print('naturalNumberList %s Iterable' % ('is' if isinstance(naturalNumberList, Iterable) else 'is not'))
print('naturalNumberList %s Iterator' % ('is' if isinstance(naturalNumberList, Iterator) else 'is not'))
print('naturalNumberIterator %s Iterable' % ('is' if isinstance(naturalNumberIterator, Iterable) else 'is not'))
print('naturalNumberIterator %s Iterator' % ('is' if isinstance(naturalNumberIterator, Iterator) else 'is not'))
for item in naturalNumberList:
        print(item)

输出结果为:

naturalNumberList is Iterable
naturalNumberList is not Iterator
naturalNumberIterator is Iterable
naturalNumberIterator is Iterator
0
1
2
3
4
5

在上例中,NaturalNumberList是可迭代对象,但不是迭代器;naturalNumberIterator既是可迭代对象,又是迭代器。这个例子中,若NaturalNumberIterator类没有__iter__方法,NaturalNumberList也能用for循环遍历,因为它有__next__方法,对于for循环已经足够了。
另外,有__next__方法而无__iter__方法的对象不能使用for循环遍历,但是可以用next方法遍历,代码如下:

it = iter([1, 2, 3, 4, 5])
while True:
    try:
        # 获得下一个值:
        x = next(it)
    except StopIteration:
        # 遇到StopIteration就退出循环
        break
小结
  • 有__iter__方法的对象都是可迭代对象,__iter__方法返回的对象必须含有__next__方法。
  • 仅有__next__方法的对象不能用for循环遍历,而可以用next方法遍历。
  • 同时有__iter__和__next__方法的对象为迭代器。
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 219,869评论 6 508
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,716评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 166,223评论 0 357
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,047评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,089评论 6 395
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,839评论 1 308
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,516评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,410评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,920评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,052评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,179评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,868评论 5 346
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,522评论 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,070评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,186评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,487评论 3 375
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,162评论 2 356

推荐阅读更多精彩内容