从日期时间np中删除nan.数组:从datetime列中提取的具有唯一值的数组

Question

我有以下列表有两个值一个是datetime.datetime(2018-06-18)和另一个NaN。两者都是从datetime列的唯一值中提取的。我只想要列表只包含日期。

# extracting date from datetime column
main_df['date'] = main_df.DateTime.dt.date 
# getting only unique values from date column
agg_hos =  main_df['date'].unique()
# output is
array([datetime.date(2018, 6, 18), NaT], dtype=object)

想要从列表中删除nan尝试了不同的网站从np数组中删除nan值

# desired output
array([datetime.date(2018, 6, 18)], dtype=object)

怎么做?

Answer 1

你可以用pd.isnull检查代替(借用这个答案):

import datetime
import numpy as np
import pandas as pd
# np.isfinite(pd.NaT), np.isnan(pd.NaT)
# -> TypeError !
arr = np.array([datetime.date(2018, 6, 18), pd.NaT])
arr = arr[~pd.isnull(arr)]
# array([datetime.date(2018, 6, 18)], dtype=object)