python之从迭代器中排除第一个元素 (Python)

我有一个生成器函数 A。

例如(实际上我有一个更复杂的函数A)，

def A(): 
    yield from [i**2 for i in range(20)] 

编写另一个生成器函数 B，我想枚举 A 返回的除第一个元素之外的所有元素。

在 Python 3 中有哪些简洁的实现方法？

请您参考如下方法:

使用itertools.islice :

itertools.islice(generator,1,None)