DataFrameでIndexが重複しているケースでの各行の特定の列の更新がうまくいかない

Question

競馬の予想ソフトを作成しているのですが、各レース毎の間隔を算出しようとしています。
indexは馬のIDとなっているので、各馬のレース分だけ重複したデータがあります。

tempDf[{'restDifference','raceName','speedScore','date'}]

休み明けを算出するロジックとして、

# 休み明け計算処理
columnsNum = tempDf.columns.get_loc('restDifference')

for horseId in tqdm(tempDf.index.unique()):
    isFirst = True
    beforeDate = ''
    count = 0
    restDifference = {}
    for index, row in tempDf[tempDf.index == horseId].sort_values('date').iterrows():
        if isFirst == True:
            isFirst = False
            tempDf[tempDf.index == horseId].sort_values('date').iat[count,columnsNum] = 0
            beforeDate = row['date']
        else:
            tempDf[tempDf.index == horseId].sort_values('date').iat[count,columnsNum] = (row['date'] - beforeDate).days
            beforeDate = row['date']
            
        count = count + 1
    
    break

aaa = tempDf[tempDf.index == 2018101626]
aaa[{'restDifference','raceName','speedScore','date'}]

restDifferenceの値が更新されなくて困ってます。
何か良い方法はありませんでしょうか？

Comments

[toast-uz]

restDifferenceを更新しようとしているコードはどの行でしょうか？見当たらないです。

[voyagerleona]

columnsNum = tempDf.columns.get_loc('restDifference')
ここでカラムの列番号をとっていて、
tempDf[tempDf.index == horseId].sort_values('date').iat[count,columnsNum] = (row['date'] - beforeDate).days

ここで入れようとしてます。
tempDf[tempDf.index == horseId].sort_values('date').iat[count,columnsNum]
で参照すると、中の値は確認できます。

[voyagerleona]

ただこのソース相当無理やり捻りだした感じのコードなので、実際どう直すのが正しい姿なのかが分かってません。。。

Answer 1

# 休み明け計算処理
columnsNum = tempDf.columns.get_loc('restDifference')

for horseId in tqdm(tempDf.index.unique()):
    isFirst = True
    beforeDate = ''
    count = 0
    restDifference = tempDf[tempDf.index == horseId].sort_values('date')
    for index, row in restDifference.iterrows():
        if isFirst == True:
            isFirst = False
            restDifference.iat[count,columnsNum] = 0
            beforeDate = row['date']
        else:
            restDifference.iat[count,columnsNum] = ((row['date'] - beforeDate).days) / 7
            beforeDate = row['date']
            
        count = count + 1
        tempDf[tempDf.index == horseId] = restDifference

これでいけました。
ただ、これだと１秒間に１回程度しか更新できないので、高速化する必要が出てきました。
別の質問で投げてみます。