python: come rimuovere i valori da 2 elenchi in base a cosa c'è nell'elenco 1

Question

Ho 2 elenchi di numeri, uno chiamato xVar e l'altro chiamato yVar. Userò questi 2 elementi per tracciare i valori X e Y su un grafico. Entrambi hanno lo stesso numero di elementi.

Normalmente, vorrei solo tracciare

ax.scatter(xVar,yVar,s=2,color='tomato'); 

Voglio rimuovere i dati dal yVar che sono oltre un certo valore, per esempio tutti i dati che hanno un valore yVar sopra 100, ma voglio anche per rimuovere il associato xVar valore. Qualcuno può suggerire un modo per creare 2 nuove variabili che rimuovano tutti i valori in yVar su 100 e i valori xVar associati ad essi? Quindi posso semplicemente sostituire xVar & nella mia linea di tracciamento con le nuove variabili.

Grazie ancora una volta tutti,

Answer 1

Ogni volta che si vuole fare qualcosa per i valori corrispondenti di due (o più elenchi, questo è ciò che è zip. Ti dà una lista, dei valori corrispondenti per ogni indice.

Quindi, in questo caso, zip le due liste insieme, poi filtrato l'elenco con zip, poi decomprimere (con zip ancora una volta, come la documentazione spiega):

xVar, yVar = zip(*((x, y) for x, y in zip(xVar, yVar) if y <= 100)) 

Se questo è fonte di confusione, mi permetta mostrano un passo alla volta:

>>> xVar = [1, 200, 300, 10] 
>>> yVar = [150, 100, 50, 200] 
>>> xyVar = zip(xVar, yVar) 
>>> xyVar 
[(1, 150), (200, 100), (300, 50), (10, 500)] 
>>> xyFiltered = [(x, y) for x, y in xyVar if y <= 100] 
>>> xyFiltered 
[(200, 100), (300, 50)] 
>>> xVar, yVar = zip(*xyFiltered) 
[(200, 300), (100, 50)] 

Answer 2

senza scrivere il codice, trovano l'indice per la y si desidera rimuovere, salvare l'indice, rimuovere il valore y, quindi rimuovere il valore x allo stesso numero di indice.

Answer 3

indici Trova che corrispondono ai dati che si vogliono mantenere:

ind = [i for i in range(len(yVar)) if yVar[i] < 100] 

e utilizzare solo quei punti:

yVar = [yVar[i] for i in ind] 
xVar = [xVar[i] for i in ind]