如何在不使用多个循环的情况下，计算与同一分组下的前一个有效数据行的天数差？

import pandas as pd

# 假设数据存储在一个名为df的DataFrame中，包含分组列group和日期列date
df = pd.DataFrame({'group': ['A', 'A', 'A', 'B', 'B', 'B'],
                   'date': ['2022-01-01', '2022-01-02', '2022-01-05', '2022-01-01', '2022-01-03', '2022-01-04']})

# 对数据进行排序，按照分组和日期的顺序排列
df = df.sort_values(['group', 'date'])

# 创建新的列diff，用于存储与前一个有效数据行的天数差
df['diff'] = 0

# 遍历数据行，计算与前一个有效数据行的天数差
for i in range(1, len(df)):
    if df.loc[i, 'group'] == df.loc[i-1, 'group']:
        diff = pd.to_datetime(df.loc[i, 'date']) - pd.to_datetime(df.loc[i-1, 'date'])
        df.loc[i, 'diff'] = diff.days

print(df)