python:如何处理if语句上的NaN数据

时间:2018-11-27 08:58:44

标签: python

我有一个csv文件,其中一些数据为空。当我使用if语句时,NaN数据将包含在else语句中。

def warna_kota(population):
    if population < 10000:
        return 'green'
    elif 10000 <= population < 20000:
        return 'orange'
    elif 20000 <= population < 30000:
        return 'brown'
    elif 40000 <= population < 50000:
        return 'yellow' 
    elif 60000 <= population < 70000:
        return 'blue'
    elif 80000 <= population < 90000:
        return 'gold'
    elif 90000 <= population < 100000:
        return 'pink'
    else:
        return 'red'

结果显示NaN数据为红色,我尝试使用它,

if population = NaN:
    return 'grey'
elif:
    ...

但是,当我尝试运行代码时,它会产生错误。 NaN数据为空,我想从NaN语句中分离出else数据。我该怎么办?

1 个答案:

答案 0 :(得分:1)

使用math.isnan(population)进行支票。

import math

# ...

if math.isnan(population):
    return 'grey'