从另一个列表中删除与nan对应的项目

Question

对于sklearn中的回归，我有一个列表x和一个列表y。不幸的是，这些列表有空格/NaN。NaN会破坏任何分析，因此我需要删除每个NaN及其在另一个列表中对应的编号（以保持x[0]和y[0]之间的关系，等等）

找到这个的优雅方法是什么？有没有其他方法可以替代循环？你知道吗

Answer 1

使用Python内置函数^{}和^{}尝试以下操作：

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]