检查日期时间列是否为空

Question

我想检查内部函数，例如datetime列是否为空做点什么。 我的样本df：

    date_dogovor    date_pogash date_pogash_posle_prodl
0   2019-03-07  2020-03-06  NaT
1   2019-02-27  2020-02-05  NaT
2   2011-10-14  2016-10-13  2019-10-13
3   2019-03-28  2020-03-06  NaT
4   2019-04-17  2020-04-06  NaT

我的功能：

def term(date_contract , date_paymnt, date_paymnt_aftr_prlngtn):
if date_paymnt_aftr_prlngtn is None:
    return date_paymnt - date_contract 
else:
    return date_paymnt_aftr_prlngtn - date_contract 

将功能应用于df：

df['term'] = df.apply(lambda x: term(x['date_dogovor'], x['date_pogash'], x['date_pogash_posle_prodl']), axis=1 )

结果错误：

df['term']

    0              NaT
1              NaT
2              NaT
3              NaT
4              NaT
            ...   
115337         NaT
115338         NaT
115339   2921 days
115340         NaT
115341         NaT
Name: term, Length: 115342, dtype: timedelta64[ns]

如何正确检查datetime列是否为空？

Answer 1

最好将numpy.where与Series.isna一起使用：

df['term'] = np.where(df['date_pogash_posle_prodl'].isna(),
                      df['date_pogash'] - df['date_dogovor'], 
                      df['date_dogovor'] - df['date_dogovor'])

您应该使用pandas.isna更改您的功能：

def term(date_contract , date_paymnt, date_paymnt_aftr_prlngtn):
    if pd.isna(date_paymnt_aftr_prlngtn):
        return date_paymnt - date_contract 
    else:
        return date_paymnt_aftr_prlngtn - date_contract