检查日期时间列是否为空

Question

我想检查内部函数，比如日期时间列是否为空，请执行某些操作。我的样本 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]

如何正确检查日期时间列是否为空？

Answer 1

这里是 better/faster 使用 numpy.where with 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

检查日期时间列是否为空

Check if datetime column is empty

python

python-datetime

pandas