Pandasデータフレームの連鎖検索

Question

私は再帰的に検索したい2つの列を持つ非常に大きなpandasデータフレームを持っています。 Iは直列OriginalIdを生成したい

NewID, OldID 
1, 0 
2, 1 
3, 2 
5, 4 
7, 6 
8, 7 
9, 5 

：

NewID, OldId, OriginalId 
1, 0, 0 
2, 1, 0 
3, 2, 0 
5, 4, 4 
7, 6, 6 
8, 7, 6 
9, 5, 4 

これは自明にソートされたデータを反復することにより、各行について解くことができる次のデータフレームの入力が与えられる

OldIdが既存のNewIdを指しているかどうかを確認し、存在する場合はその行のOriginalIdをOriginalIdに設定します。

これは、以下のアルゴリズムにより、繰り返しの列をマージして更新することで解決することができます：累積和または類似経由でこれを行うにはパンダに優しい方法があるべきよう

Merge OldId to NewId. 
For any one that did not match, set OriginalId to OldId. 
If they did match, set OldId to OldId for the matched column. 
Repeat until OriginalIds are all filled in. 

を感じています。

Answer 1

簡単：

df.set_index('NewID', inplace=True) 
df.loc[:, 'OriginalId'] = df.loc[df['OldId'], 'OldID'].fillna(df['OldId'])