从Python列表中删除nan值

时间:2017-02-04 13:37:55

标签: python arrays numpy nan

有人能看出为什么这不起作用?我试图从我的python List /数组中删除nan值。

import math
import numpy as np

def clean_List_nan(List):
    Myarray=np.array(List)
    x = float('nan')
    for elem in Myarray:
        if math.isnan(x):
            x = 0.0
    return Myarray


oldlist =[nan, 19523.3211203121, 19738.4276377355, 19654.8478302742, 119.636737571360, 19712.4329437810, nan, 20052.3645613346, 19846.4815936009, 20041.8676619438, 19921.8126944154, nan, 20030.5073635719]

print(clean_List_nan(oldlist))

2 个答案:

答案 0 :(得分:7)

您的函数中的控制流没有意义 - 您将变量x设置为nan,然后检查循环中是否确实nan并将其设置为0你从不接触或检查阵列的任何元素。

要将nan值正确转换为0,您只需使用numpy.nan_to_num,因为您似乎正在使用NumPy数组。

<强>演示

In[37]: arr
Out[37]: 
array([            nan,  19523.32112031,  19738.42763774,  19654.84783027,
          119.63673757,  19712.43294378,             nan,  20052.36456133,
        19846.4815936 ,  20041.86766194,  19921.81269442,             nan,
        20030.50736357])

In[38]: np.nan_to_num(arr)
Out[38]: 
array([     0.        ,  19523.32112031,  19738.42763774,  19654.84783027,
          119.63673757,  19712.43294378,      0.        ,  20052.36456133,
        19846.4815936 ,  20041.86766194,  19921.81269442,      0.        ,
        20030.50736357])

如果您对常规Python列表的方法的功能版本更感兴趣,您可以尝试这样的事情,或者像fafl提供的列表理解。

In[39]: list(map(lambda x: 0.0 if math.isnan(x) else x, oldlist))
Out[39]: 
[0.0,
 19523.3211203121,
 19738.4276377355,
 19654.8478302742,
 119.63673757136,
 19712.432943781,
 0.0,
 20052.3645613346,
 19846.4815936009,
 20041.8676619438,
 19921.8126944154,
 0.0,
 20030.5073635719]

答案 1 :(得分:2)

米奇的答案可能是最好的方法。如果您希望手动编写,可以执行类似

的操作
cleanlist = [0.0 if math.isnan(x) else x for x in oldlist]