Discussion :

[nastassia91]

Bonjour,

Je suis entrain de manipuler deux listes de listes de même taille.
Pour la première, je veux supprimer les éléments négatifs.
Pour la deuxième, je veux supprimer les éléments ayant les même indices que les éléments supprimés dans la première liste.

Un exemple:

Si

Code :

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

1
2
A = [ [1,-1,2,4], [2,-2,-4,-5], [0,2,-5,4] ]
B = [ [1,0,-1,5], [6,7,10,4],  [-5,7,2,1] ]

Le résultat sera

Code :

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

1
2
3
 
A = [ [1,2,4], [2], [0,2,4] ]
B = [ [1,-1,5], [6], [-5,7,1] ]

J'ai commencé la première étape mais je n'arrive pas à supprimer les éléments dans la deuxième:

Code :

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

1
2
3
 
for i in range(len(A)):
    A[i] = [x for x in A[i] if x >=0 ]

J'espère que vous pouvez m'aider
Merci

[VinsS]

Salut,

Tu parcours les deux listes ensemble et tu crées d'abord la liste des index pour les supprimer avec pop.

Code :

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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
 
>>> A = [ [1,-1,2,4], [2,-2,-4,-5], [0,2,-5,4] ]
>>> B = [ [1,0,-1,5], [6,7,10,4],  [-5,7,2,1] ]
>>> for a, b in zip(A, B):
...     idxs = [a.index(x) for x in a if x < 0]
...     for idx in reversed(idxs):
...             a.pop(idx)
...             b.pop(idx)
... 
-1
0
-5
4
-4
10
-2
7
-5
2
>>> A
[[1, 2, 4], [2], [0, 2, 4]]
>>> B
[[1, -1, 5], [6], [-5, 7, 1]]
>>>

[nastassia91]

Génial !!
Merci beaucoup pour votre réponse !!