Discussion :

[DonutMan]

Bonjour,

soit les deux listes suivantes :

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
a = [0,0,1,0,1,0]
b = [1,2,3,4,5,6]

Je souhaite extraire les éléments de b pour lesquels a est positif.
Autrement dit avoir in fine la liste [3,5]

Ceci ne marche pas :

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

c = [v for v in b if a>0]

En rusant un peu, je me dépatouille avec ce code là qui donne bien le résultat escompté :

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

c = [v for (n,v) in enumerate(b) if a[n]>0]

Néanmoins, je ne suis pas sûr que ça soit très propre... Y'a-t'il plus "pythonic" comme solution ?

Idéalement j'aimerais rester en python pur mais s'il existe une solution élégante avec un module, je suis preneur !
Par exemple, j'avais pensé à numpy mais np.where() ne donne pas ce que je souhaite. Il me semble que np.where renverra toujours un vecteur de la même taille que ses arguments...

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

np.where(np.array(a)>0, b, None)

Merci d'avance pour vos retours éclairés

D.

[Sve@r]

Bonjour

Ceci ne marche pas : c = [v for v in b if a>0]

Oui c'est évident. Déjà rien que pour toi, qu'est censé signifier "si une liste est supérieure à 0"??? Alors pour Python...
Il faut bien prendre conscience qu'une expression "en intension" n'est que l'écriture simplifiée d'une boucle "for"...

Code python :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
final=[]
for x in liste:
	if condition: final.append(x)

... qui se réécrira alors final=[x for x in liste if condition]. La condition, elle, ne change pas. Donc si la condition est mauvaise dès le départ, ce n'est pas en la réécrivant dans l'expression simplifiée que ça la rendra correcte.

Néanmoins, je ne suis pas sûr que ça soit très propre... Y'a-t'il plus "pythonic" comme solution ?

Déjà question propreté il sera bon de se demander si la liste se justifie pour un truc qui n'évolue pas (pour ça il y a le tuple).
Ensuite puisque ton code présuppose "a" et "b" de même taille, alors il y a le zip... c=tuple(y for (x, y) in zip(a, b) if x > 0)...

[BufferBob]

salut,

avec numpy on peut faire un truc dans ce goût là, sans dire que c'est mieux/plus pythonic :

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
a = np.array([0,0,1,0,1,0], dtype='bool')
b = np.array([1,2,3,4,5,6])
b[a]

[Sve@r]

sans dire que c'est mieux/plus pythonic

Je sais pas si c'est plus/moins pythonic mais en tout cas terminer par un simple b[a] c'est classe !!!
Faut vraiment que je me mette à numpy moi...

[DonutMan]

Bonjour,
merci à vous pour vos retours rapides
J'ai modifié mon code en conséquence

A bientôt,

D.

[Invité]

Faut vraiment que je me mette à numpy moi...

J'utilise très superficiellement numpy avec cv2, qui sont les fous qui ont créé ce truc imbitable et inutilisable avec des boucles et des conditions ???

[wiztricks]

J'utilise très superficiellement numpy avec cv2, qui sont les fous qui ont créé ce truc imbitable et inutilisable avec des boucles et des conditions ???

numpy et cv2 sont des bibliothèques spécialisées et optimisées (écrites en C).

On peut faire ce genre de choses sur des domaines "stables" (côté algorithmes) pour lesquels nombre de cas d'utilisation (ce qu'on en attend) sont bien définis.

La difficulté lorsqu'on veut utiliser ces joujoux est d'arriver à se mettre dans la peau de ceux pour qui il a été destiné pour comprendre comment sont construites les solutions proposées.

On a un peu le même soucis avec le Python de base. Au début on fait des exercices pour apprendre boucles, conditions, listes,... puis lorsqu'on en a marre de répéter les mêmes constructions, on va trouver une bibliothèque standard, une fonction qui... Et il n'y a rien de plus frustrant pour le débutant qui patauge à écrire des trucs simples de voir que les plus avancés utiliseront des "trucs" qu'il ignore.

- W