pandas: 条件を指定して行を抽出後、値を変更する。(python)

Question

ID   ruiseki
001  2 
001  3
002  4
002  5
001  6
001  7
003  8

このようなデータテーブルがあったとして、idごとに、その１つ前のデータのruisekiを引いた値をカラムに追加したいと考えています。つまり下のようなデータテーブルに変形したいと考えています。

ID   ruiseki  target 
001  2        0
001  3        1
002  4        0
002  5        1
001  6        3
001  7        1
003  8        0

現在、以下のように、IDごとにデータを抽出して、その中で現在の行の１つ前の行を参照し、そのruisekiの値との差をtargetに入れていくような方法を考えているのですが、うまくいきません。
どうしたらうまくいくでしょうか？
回答よろしくお願いします。

for i in df["ID"].unique():
  for j in range(0,len(df[df["ID"]==i])):
    if(j==0):
        df[df["ID"]==i].iloc[0,2]=0
    else:
        df[df["ID"]==i].iloc[j,2]=df[df["ID"]==i].iloc[j,1]-df[df["ID"]==i].iloc[j-1,1]

Answer 1

以下の手順でできます。

1. groupby("ID") で ID ごとにグループ化する。
2. diff() で1つ前の値との差分をとる (グループごと)
3. diff() の結果、一番最初の値は1つ前の値との比較はできないので、NaN になるが、これを fillna() で 0 埋めする。

ID,ruiseki
001,2 
001,3
002,4
002,5
001,6
001,7
003,8

import pandas as pd

df = pd.read_csv("test.csv")
df["target"] = df.groupby("ID").diff().fillna(0, downcast="infer")
print(df)
#    ID  ruiseki  target
# 0   1        2       0
# 1   1        3       1
# 2   2        4       0
# 3   2        5       1
# 4   1        6       3
# 5   1        7       1
# 6   3        8       0

Comments

[pdaocmhii]

助かりました。
ありがとうございます！