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

复制代码

永恒的蓝色梦想

1. class head:
   
2.     def __init__(self,iterable,n,/):
   
3.         if isinstance(n,int) and n>=0:
   
4.             self.__n=n
   
5.         else:
   
6.             raise TypeError(f"'{n.__class__.__name__}' object cannot be interpreted as an integer")
   
7. 
   
8.         self.__it=iter(iterable)
   
9. 
   
10.     def __iter__(self):
    
11.         return self
    
12. 
    
13.     def __next__(self):
    
14.         if self.__n:
    
15.             self.__n-=1
    
16.             return self.__it.__next__()
    
17. 
    
18.         raise StopIteration

复制代码