Concaténer tout en générant

Question

Comment puis-je concaténer deux éléments en générant à partir d'une fonction en python?

Le cas de base:

import itertools 

def test(): 
    for number in range(0,10): 
     yield number 

list(test()) # [0...9] 

si je veux donner à la fois la number et sa place number**2

import itertools 

def test(): 
    for number in range(0,10): 
     yield itertools.chain.from_iterable([ 
      number,number*2 
     ]) 

list(test()) 

# [0,0,1,1,2,4,3,9,...] pretended 
# <itertools.chain at 0x135decfd0> ... what I got 

faire Cependant itertools.chain.from_iterable([generator1, generator2]) de l'extérieur donne le résultat attendu.

def first_list(): 
    for number in range(0,5): 
     yield number 

def second_list(): 
    for number in range(5,10): 
     yield number 

list(itertools.chain.from_iterable([first_list(), second_list()])) 

Answer 1

A manière mple est:

def test(): 
    for number in range(0,10): 
     yield number 
     yield number**2 

Answer 2

si je veux donner à la fois le nombre et le nombre carré ** 2

Vous pouvez simplement donner un tuple, comme dans

yield number, number ** 2 

Par exemple:

In [1]: import itertools 
    ...: 
    ...: def test(): 
    ...:  for number in range(0,10): 
    ...:   yield number, number ** 2 
    ...:  
    ...: 
    ...: list(test()) # [0...9] 
    ...: 
Out[1]: 
[(0, 0), 
(1, 1), 
(2, 4), 
(3, 9), 
(4, 16), 
(5, 25), 
(6, 36), 
(7, 49), 
(8, 64), 
(9, 81)] 

Answer 3

Python 3.3 a également introduit le yield from (voir aussi PEP-380) syntaxe qui vous permet de faire quelque chose comme ceci:

>>> def test(): 
... for number in range(10): 
...  yield from (number, number**2) 
... 
>>> list(test()) 
[0, 0, 1, 1, 2, 4, 3, 9, 4, 16, 5, 25, 6, 36, 7, 49, 8, 64, 9, 81]