compréhension de liste paire par paire

Version imprimable

Petite question super basique sur laquelle je m'arrache les cheveux :

Peut-on générer une liste par compréhension en ajoutant des éléments 2 par 2 (n par n) ?
Exemple :
Code:

1 2 3 4 5 6 7 8 9 10 11 >>> [ a for a in xrange( 5 )] [0, 1, 2, 3, 4] >>> [ (a,a+1) for a in xrange( 5 )] [(0, 1), (1, 2), (2, 3), (3, 4), (4, 5)] >>> [ a,a+1 for a in xrange( 5 )] File "<stdin>", line 1 [ a,a+1 for a in xrange( 5 )] ^ SyntaxError: invalid syntax
Première ligne : génération en ajoutant les éléments 1 par 1 : OK
Deuxième ligne : génération en ajoutant les éléments 2-uplet par 2-uplet : OK
Troisième ligne : génération en ajoutant les éléments 2 par 2 : fail !
Mais comment obtenir [0, 1, 1, 2, 2, 3, 3, 4, 4, 5] (par compréhension de liste bien sûr) ? :cry:

Merci d'avance !

Bonjour,

Code:

1
2
3
 
print [x//2 for x in xrange(1,20)]
[0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9]

Tyrtamos

Salut,

juste une idée:
Code:

1 2 3 max = 5 tata = [x for x in [x/2 for x in range(max *2 + 2)]][1:-1] print tata
A+

Pfeuh

Grillé! :oops:

Merci de vos réponses, mais hum...
je n'ai pas réussi à me faire comprendre... désolé...

En fait, ce n'est pas la suite 0, 1, 1, 2, 2, 3, 3, 4, 4, 5 qui m’intéresse, c'est la façon de générer des listes en ajoutant les éléments 2 par 2.

Voici une variante du code :
Code:

1 2 3 4 5 >>> f1 = lambda x : # fonction mystère >>> f2 = lambda x : # autre fonction mystère >>> f3 = lambda x : (f1( x ),f2( x )) >>> [f3( x ) for x in xrange( 5 )] [(0, -11), (-2, -8), (2, -3), (18, 4), (52, 13)]
Et bien sûr, ce que je voudrais obtenir c'est [0, -11, -2, -8, 2, -3, 18, 4, 52, 13]...

Le cas général qui m’intéresse, c'est avec une fonction f3 qui renvoie un n-uplet quelconque.
J'avais bien pensé à

Code:

>>> [*f3( x ) for x in xrange( 5 )]

mais ça ne marche pas...

06/12/2010, 16h53
josmiley

Code:

sum([(a,a+1) for a in xrange( 5 )],())
06/12/2010, 17h12
tyrtamos

Je trouvais la 1ère question limpide, mais je ne comprends pas la seconde.

Est-ce que tu veux savoir comment on passe de [(0, -11), (-2, -8), (2, -3), (18, 4), (52, 13)] à [0, -11, -2, -8, 2, -3, 18, 4, 52, 13]???

Tyrtamos

Personnellement, je n'essaierais pas de le faire par compréhension de liste. Avec des listes 'conventionnelles', c'est plus facile et plus clair

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
def f1(i):
    return i
 
def f2(i):
    # square
    return i * i
 
def f3(i):
    # cube
    return i * i * i
 
def createList(indexes, hooks):
    retlist = []
    for index in indexes:
        for hook in hooks:
            retlist.append(hook(index))
    return retlist
 
print createList(range(10), [f1, f2, f3])

06/12/2010, 19h05
sopsag

Merci josmiley, ta solution avec sum est celle qui se rapproche le plus de ce que je cherchais.

Toutefois je regrette qu'une syntaxe à la

Code:

[*f3( x ) for x in xrange( 5 )]

ne fonctionne pas. Ça me paraîtrait logique...

Le pire, c'est que j'avais déjà posté une question de ce genre il y a quelques mois... :oops:
06/12/2010, 19h12
sopsag

En fait, pfeuh, ta super fonction createList peut se résumer à :

Code:

sum( [[f1(x),f2(x),f3(x)] for x in xrange( 10 )],[] )

Personnellement, je trouve ça plus clair... mais évidemment, tous les goûts sont dans python :mouarf:
06/12/2010, 19h48
pfeuh

Citation:

Envoyé par sopsag

En fait, pfeuh, ta super fonction createList peut se résumer à :

Code:

sum( [[f1(x),f2(x),f3(x)] for x in xrange( 10 )],[] )

Personnellement, je trouve ça plus clair... mais évidemment, tous les goûts sont dans python :mouarf:

Du coup, je viens de découvrir la fonction sum 8-)
06/12/2010, 22h06
alexdevl
Bonjour,

Voila une solution (j'utilise plutôt les indices de la liste que les éléments mais bon...)
Code:

1 2 l=[(0, -11), (-2, -8), (2, -3), (18, 4), (52, 13)] print [l[t][e] for t in range(len(l)) for e in [0,1]]
ce qui fait :

Code:

[0, -11, -2, -8, 2, -3, 18, 4, 52, 13]

Une fonction générale étant :

Code:

[l[t][e] for t in range(len(l)) for e in range(len(l[0]))]

Merci alexdev, encore une syntaxe que je ne connaissais pas et qui me semble plus judicieuse que sum() :ccool: En plus, c'est très compréhensible.
Code:

1 2 3 4 5 6 7 8 9 10 11 12 13 def f1(i): # double return i * 2 def f2(i): # square return i * i def f3(i): # cube return i * i * i print [f(i) for i in range(10) for f in [f1, f2, f3]]

Bonjour,

S'il s'agit de "mettre à plat" une liste de tuples, je préfère pour ma part utiliser une fonction générale de mise à plat d'un arbre (une liste de listes ou de tuples quelque soit sa profondeur):
Code:

1 2 3 4 5 6 7 8 9 def aplatliste(L): R = [] for elem in L: if isinstance(elem, (list, tuple)): R.extend(aplatliste(elem)) else: R.append(elem) return R
Par exemple:
Code:

1 2 3 4 L = [1,2,3,(4,5,(8,9),6),7] print aplatliste(L) [1, 2, 3, 4, 5, 8, 9, 6, 7]
Et pour revenir au cas traité ici:
Code:

1 2 3 4 L = [(0, -11), (-2, -8), (2, -3), (18, 4), (52, 13)] print aplatliste(L) [0, -11, -2, -8, 2, -3, 18, 4, 52, 13]
Tyrtamos

Edit: sinon, en se limitant à la liste de tuples donnée, la solution la plus courte en list compréhension me semble être:
Code:

1 2 3 4 L = [(0, -11), (-2, -8), (2, -3), (18, 4), (52, 13)] print [x for T in L for x in T] [0, -11, -2, -8, 2, -3, 18, 4, 52, 13]

07/12/2010, 21h36
alexdevl

Ha très bien il me semblait qu'il existait quelque chose comme ceci
Mais je n'arrivais pas à trouver tout simplement

Code:

[x for T in L for x in T]

,
J'inversais les ''for''..
Merci à Tyrtamos:D

Alex