Pandas Dataframe, совпадающий с датами между строками

Question

REF         FROM_DATE               EXEMPT_FROM
10001033    2017-07-11 00:00:00     2017-07-11 00:00:00
10001033    2017-07-11 00:00:00     2017-08-10 00:00:00
10001033    2017-07-11 00:00:00     2018-04-01 00:00:00
10001769    2018-05-15 00:00:00     2016-07-15 00:00:00
10001769    2018-05-15 00:00:00     2018-01-03 00:00:00
10001769    2018-05-15 00:00:00     2018-05-15 00:00:00
10001769    2018-05-15 00:00:00     2018-06-14 00:00:00

У меня есть вышеуказанный фрейм данных, и я хотел бы иметь возможность подсчитать количество уникальных REF, первая строка которых From_DATE не совпадает с первой строкой Exempt_FROM.

Например, первый номер REF 10001033 FROM_DATE и EXEMPT_FROM совпадают, а второй REF 10001769 - нет.Какой может быть лучший подход?

Answer 1

IIUC, это то, что вы хотите

A = df.groupby('REF').first()
A.FROM_DATE == A.EXEMPT_FROM

Вывод:

REF
10001033     True
10001769    False

Если вы хотите узнать, сколько раз у вас есть желаемое совпадение, примените sum

sum(A.FROM_DATE == A.EXEMPT_FROM)