Python 实现 head()

zltzlt

Python 实现 head()

语法

1. head(iterable, n)

复制代码

描述

以生成器的形式返回 iterable 的前 n 个元素。如果 iterable 中的元素个数 < n，则返回 iterable 中的所有元素。

参数

参数	描述
iterable	可迭代对象。
n	必须是整数。

返回值

生成器。

例子

1. >>> def gen():
   
2.         yield 1
   
3.         yield 2
   
4.         yield 3
   
5.         yield 4
   
6.         yield 5
   
7. 
   
8.        
   
9. >>> g1 = gen()
   
10. >>> list(head(g1, 3))
    
11. [1, 2, 3]
    
12. >>> g2 = gen()
    
13. >>> list(head(g2, 1))
    
14. [1]
    
15. >>> g3 = gen()
    
16. >>> list(head(g3, 8))
    
17. [1, 2, 3, 4, 5]

复制代码

代码

1. def head(iterable, n):
   
2.     """
   
3.     head(iterable, n) -> [generator object]
   
4.     以生成器的形式返回 iterable 的前 n 个元素
   
5.     """
   
6.     it = iter(iterable)
   
7.     for i in range(n):
   
8.         try:
   
9.             yield next(it)
   
10.         except StopIteration:
    
11.             break

复制代码