Удаление элементов, соответствующих nan в одном списке из другого списка

Question

Для регрессий в sklearn у меня есть список x и список y. К сожалению, у этих списков есть пробелы/NaN. NaN сбой любого анализа, поэтому мне нужно удалить каждый NaN и его соответствующий номер в другом списке (чтобы сохранить связь между и y[0] и т. Д.)

Что такое элегантный способ найти это? Есть ли альтернатива петлям?

Answer 1

Попробуйте следующее, используя Python встроенную функцию zip и filter:

def filter_nans(x, y): 
    filtered = filter(lambda o: not math.isnan(o[0]) and not math.isnan(o[1]), zip(x, y))  
    return [el[0] for el in filtered], [el[1] for el in filtered] 

Так, например:

x = [1, 2, 3, 4, 5, 6, None, 8] 
y = [10, 'not a number', 30, 40, 50, 60, 70, 80] 

x, y = filter_nans(x, y) 

>>> x 
[1, 3, 4, 5, 6, 8] 
>>> y 
[10, 30, 40, 50, 60, 80]