Как сравнить столбец даты с датой и добавить текстовый столбец?

Question

У меня есть следующий фрейм данных:

id      date
1       2019-09-28
2       NaT
3       2017-09-28

Я хочу сравнить столбец даты с датой: 2018-09-28.Затем добавьте статус столбца, который добавляет строку «Больше» или «Меньше» на основе сравнения.Таким образом, результат будет примерно таким:

id      date          status
1       2019-09-28    Greater
2       NaT
3       2017-09-28    Less

Answer 1

С сериями Pandas datetime вы можете сравнить значения со строкой:

df['date'] = pd.to_datetime(df['date'])

df['status'] = np.where(df['date'] >= '2018-09-28', 'Greater', 'Less')
df.loc[df['date'].isnull(), 'status'] = ''

print(df)

   id       date   status
0   1 2019-09-28  Greater
1   2        NaT         
2   3 2017-09-28     Less