Pandasでデータ抽出挿入がうまくできない

Question

Pandasでデータ抽出挿入がうまくできないので教えていただけないでしょうか。

実現したいこと

下の表でCondがTrueの場合にOffset値だけシフトしてPrice値を入れた列を追加したい。

例えば、行2（Cond=True）の場合、Price 300 を 2つOffsetした行0に300を挿入
行3（Cond=True）の場合、Price 400 を 1つOffsetした行2に400を挿入

import pandas as pd

df = pd.DataFrame({'Price': [150, 250, 300, 400, 500,550],
                   'Cond': [False, False, True, True, None, True],
                   'Offset': [None, None, 2, 1, None, 1]})
----------
 	Price 	Cond 	Offset
0 	150 	False 	NaN
1 	250 	False 	NaN
2 	300 	True 	2.0
3 	400 	True 	1.0
4 	500 	None 	NaN
5 	550 	True 	1.0

望む結果は次の通りです。

 	Price 	Cond 	Offset	NEW
0 	150 	False 	NaN　　　300
1 	250 	False 	NaN　　　
2 	300 	True 	2.0　　　400
3 	400 	True 	1.0
4 	500 	None 	NaN　　　550
5 	550 	True 	1.0

試したこと

以下のように5段階で処理しようと思いましたが最後のｅの処理がわかりません。
a = df[(df['Cond']==True)].index #条件にあてはまるindex抽出
b = df.loc[(df['Cond']==True),'Offset'].astype("int64") #条件にあてはまるOffset抽出
c = a-b #新たなindex
d = df.loc[(df['Cond']==True),'Price'] #新たなPrice
e = d.reindex(c.index) #ｃとｄの結果から新たなindex/Priceにしたい

どうかよろしくお願いいたします。

Answer 1

def set_new_price(r):
    if r['Cond']:
        df.loc[r.name - r['Offset'], 'New'] = r['Price']
df.apply(set_new_price, axis=1)

print(df)

#
   Price   Cond  Offset    New
0    150  False     NaN  300.0
1    250  False     NaN    NaN
2    300   True     2.0  400.0
3    400   True     1.0    NaN
4    500   None     NaN  550.0
5    550   True     1.0    NaN

Comments

[morimori_4]

回答いただき感謝しております。解決できました。