Python 基于dataframe的各个独立部分中的连续行创建特征

我想不出更好的方法来表达这个问题，如果你有更好的方法，请编辑

日期 国家 个案总数 1 丹麦 2 2 丹麦 5 三 丹麦 10 4 丹麦 17 1 美国 5 2 美国 13 三 美国 23 4 美国 37 1 印度 0 2 印度 2 三 印度 9 4 印度 40 使用：

或：

---

你认为它也可以和透视表一起工作吗？只是好奇。我试过了，但似乎无法得到它。@callmeaningyouwant-我认为不是，“diff”不是聚合函数

df['new_cases'] = df['total_cases'] - df.groupby('country')['total_cases'].shift(1)

df['new_cases'] = df.groupby('country')['total_cases'].diff(1)