Сравнение двух столбцов даты, чтобы указать, произошла ли запись после другой ⇐ Python

[Anonymous]

У меня есть фрейм данных, в котором я хочу вернуть количество (долю) пациентов, которые прошли последующее наблюдение после постановки диагноза заболевания.
Исходный DF (пример 1 пациента)< /p>

Код: Выделить всё

| patient_id | app_date   | diag_date  | cancer_yn |
|------------|------------|------------|-----------|
| 1          | 2024-01-11 | NaT        | NaN       |
| 1          | 2024-03-14 | 2024-03-14 | 1         |
| 1          | 2024-04-09 | NaT        | NaN       |
| 1          | 2024-09-09 | NaT        | NaN       |

Промежуточный DF (указывает, что запись о приеме каждого пациента соответствовала дате постановки диагноза или нет)

Код: Выделить всё

| patient_id | app_date   | diag_date  | cancer_yn | fup_yn |
|------------|------------|------------|-----------|--------|
| 1          | 2024-01-11 | NaT        | NaN       | 0      |
| 1          | 2024-03-14 | 2024-03-14 | 1         | 0      |
| 1          | 2024-04-09 | NaT        | NaN       | 1      |
| 1          | 2024-09-09 | NaT        | NaN       | 1      |

Суммарный DF (свернутый, по группам по Patient_id и value_counts() или что-то похожее

Код: Выделить всё

| patient_with_fup | count |
|------------------|-------|
| 1                | 24    |
| 0                | 67    |

Вы можете видеть, что 2-я запись указывает на прием, на котором был поставлен диагноз (

Код: Выделить всё

diag_date

доступен и совпадает с app_date), у этого пациента были последующие приемы, я хочу отметить, что это так (скажем, Follow_ups == 1.
Мне трудно понять, как можно группировать разных пациентов и применять value_counts() к флагу, указывающему, что у пациента были последующие осмотры после диагностического приема. Предложения по основанному способу изменения формы данных и генерации. флаг был бы великолепен.

Подробнее здесь: https://stackoverflow.com/questions/791 ... er-another