Я хочу вставить новый столбец в пандах с последующим условием группировки

Question

введите описание изображения здесь Я хочу добавить столбец для продолжительности, который вычитает максимальную дату из минимальной даты для каждой компании (ComapnyName)

Answer 1

Использование:

m1 = df.groupby('CompanyName')['Date'].transform('max')
m2 = df.groupby('CompanyName')['Date'].transform('min')
df['duration'] = (m1 - m2).dt.days