python - 在 pandas 数据框中查找元素的最新值

Question

我有一个这样的数据框

HOME    AWAY    COUNT_H    COUNT_A    N_H     N_A
Wild    Sharks    1          1        0.5    -0.4
Jets    Sabres    1          1        0.7     0.3
Sharks  Wild      2          2        0.2    -0.1
Jets    Stars     2          1       -0.2     0.4
Stars   Wild      1          3        0.7     0.3
Sharks  Sabres    3          2        0.5    -0.4

我需要创建两个新列“K_H”和“K_A”，它们显示当计数比当前行指示的数少一个时“N_H”和“N_A”是什么。

例如，我想要的输出是这样的：

HOME    AWAY    COUNT_H    COUNT_A    N_H     N_A    K_H     K_A
Wild    Sharks    1          1        0.5    -0.4    n/a     n/a
Jets    Sabres    1          1        0.7     0.3    n/a     n/a
Sharks  Wild      2          2        0.2    -0.1    -0.4    0.5
Jets    Stars     2          1       -0.2     0.4    0.7     n/a
Stars   Wild      2          3        0.7     0.3    0.4    -0.1
Sharks  Sabres    3          2        0.5    -0.4    0.2     0.3

我在这里很迷茫。我试图在创建新列时对数据框进行切片，但这不起作用。我也做了一个 for 循环，但速度慢得让人难以忍受。我可以在 Excel 中使用 INDEX(MATCH()) 函数很容易地做到这一点，但我真的需要熊猫的东西。

任何帮助表示赞赏

谢谢

标签： pythonpandas