Python 小技巧 069：iter() 函数的另一个用法

zltzlt

iter() 函数除了获取可迭代对象的迭代器，还有另一个用法。

看一下帮助文档：

1. >>> help(iter)
   
2. Help on built-in function iter in module builtins:
   
3. 
   
4. iter(...)
   
5.     iter(iterable) -> iterator
   
6.     iter(callable, sentinel) -> iterator
   
7.    
   
8.     Get an iterator from an object.  In the first form, the argument must
   
9.     supply its own iterator, or be a sequence.
   
10.     In the second form, the callable is called until it returns the sentinel.

复制代码

它的第二种形式是：

1. iter(callable, sentinel) -> iterator

复制代码

返回的也是一个迭代器。当函数 callable 调用的结果不为 sentinel 时，返回调用结果。否则退出循环。

iter() 函数可以在询问用户输入多行时用到：

1. print("请输入多行文本（连按两次 <Enter> 键退出）：")
   
2. res = []
   
3. for i in iter(input, ''):
   
4.     res.append(i)
   
5. print("你输入的文本是：")
   
6. print("\n".join(res))

复制代码

运行结果：

1. 请输入多行文本（连按两次 <Enter> 键退出）：
   
2. asdasdasd
   
3. fghfghfgh
   
4. 
   
5. 你输入的文本是：
   
6. asdasdasd
   
7. fghfghfgh

复制代码

iter() 函数还能这样用：

1. n = 0
   
2. 
   
3. 
   
4. def f():
   
5.     global n
   
6.     n += 1
   
7.     return n
   
8. 
   
9. 
   
10. for i in iter(f, 5):    # 当 f() 不为 5 时继续循环
    
11.     print(i)

复制代码

运行结果：

1. 1
   
2. 2
   
3. 3
   
4. 4

复制代码

_2_

已经了解！@zltzlt

永恒的蓝色梦想

新语法：

1. res = []
   
2. while(i:=input()):
   
3.     res.append(i)

复制代码

_2_

永恒的蓝色梦想 发表于 2020-4-3 13:54
新语法：

:= 说：跟我们海象有啥关系

_2_

永恒的蓝色梦想 发表于 2020-4-3 13:54
新语法：

加一句提示：

1. res = []
   
2. print("Press <Enter> to exit.")
   
3. while(i:=input()):
   
4.     res.append(i)

复制代码