Как группировать значения на основе условий временных меток? ⇐ Python

[Anonymous]

У меня есть фрейм данных. Выглядит это так:

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

prod_type                timestamp1                  timestamp2
a1                     2023-12-02                    2023-12-01
a2                     2023-10-10                    2023-09-02
a1                     2023-12-11                    2023-12-22
a3                     2023-05-11                    2023-06-21
.......

Мне нужно назначить prod_type новой группе (новый параметр group_id) с распределением дат по timestamp1, не превышающим месяц (поэтому разница между max и min внутри group_id должна быть не выше, чем 30 дней). Аналогичным образом распределение дат в timestamp2 не должно превышать месяца (поэтому разница между максимальным и минимальным значением внутри group_id не должна превышать 30 дней). Цель состоит в том, чтобы максимизировать среднее количество prod_id на один group_id.
Я пробовал это, но все равно получаю group_id с распределением дат в timestamp1 более чем на 30 дней:

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

# Define function to assign group_id
def assign_group(row):
group_id = 0
for i, (index, group) in enumerate(df.groupby('group_id')):
if (row['timestamp1'] - group['timestamp1'].min()).days 

Подробнее здесь: [url]https://stackoverflow.com/questions/78300451/how-to-group-values-based-on-timestamps-conditions[/url]