предложение или с датой и нулевыми значениями в python

Question

У меня следующий df:

   date_from    date_to      birth_date    death_date
0  2016-01-10   2019-06-05   2015-02-15    2018-07-25
1  2016-05-11   2020-06-13   2014-03-07    2020-07-11
2  2016-02-23   Nat          2014-03-07    2019-06-08
3  2015-12-08   Nat          2014-03-07    2019-06-08

Я пытаюсь выбрать все случаи, когда date_to> death_date ИЛИ где date_to = Nat.

Я пробовал следующий код:

df = df[(df['date_to'] > df['death_date']) | (df[df['DATE_TO'].isnull()])]

, но я получаю следующее сообщение об ошибке

'TypeError: cannot compare a dtyped [float64] array with a scalar of type [bool]'

, и я действительно не знаю, как обойти эту проблему.

Answer 1

Из вашего вопроса

import pandas as pd
# ..... your data frame df ......

# considering that you have the following types

>>> df.dtypes
date_from     datetime64[ns]
date_to       datetime64[ns]
birth_date    datetime64[ns]
death_date    datetime64[ns]
dtype: object  

df = df[(df['date_to'] > df['death_date']) | (df['date_to'].isnull())]

>>> df
date_from    date_to birth_date death_date
0 2016-01-10 2019-06-05 2015-02-15 2018-07-25
2 2016-02-23        NaT 2014-03-07 2019-06-08
3 2015-12-08        NaT 2014-03-07 2019-06-08

Если столбец date_to не является datetime, вы можете преобразовать его следующим образом

df['date_to'] = df['date_to'].replace('Nat', pd.NaT)
df['date_to'] = pd.to_datetime(df['date_to'])