行数が異なるデータフレームの内積計算方法について

Question

お世話になります。

以下の行数が異なるデータフレーム（Ｂ列は９行、Ｃ列は２行）があり、Ｃ列を１行ずつずらしながら、Ｂ列とＣ列の内積結果をＤ列に出力したいです。

具体的にはＢ列の１～２行目とＣ列の内積（11111+22222）を新たに作るＤ列の１行目に出力、次にＢ列の２～３行目とＣ列の内積（22111+33222）をＤの２行目・・・、それを最終行（計算可能な行）まで繰り返し行いたいのですが、方法が思いつきません。

ご教示お願いいたします。

|Index|Ｂ|Ｃ|Ｄ|
|:--|:--:|:--:|
|1|11|111|『以下出力先』
|2|22|222|
|3|33|NaN|
|4|44|NaN|
|5|55|NaN|
|6|66|NaN|
|7|77|NaN|
|8|88|NaN|
|9|99|NaN|

Answer 1

とりあえず、 DataFrame.rolling().apply() を使って書いてみました

import pandas as pd
import numpy as np
df = pd.DataFrame({
    'B': [11,22,33,44,55,66,77,88,99],
    'C': [111,222,np.nan,np.nan,np.nan,np.nan,np.nan,np.nan,np.nan]
})
print(df)
#    B      C
#0  11  111.0
#1  22  222.0
#2  33    NaN
#3  44    NaN
#4  55    NaN
#5  66    NaN
#6  77    NaN
#7  88    NaN
#8  99    NaN

df['D'] = df['B'].rolling(2).apply(lambda d: (d*df.loc[0:1,'C']).sum()).shift(-1)
print(df)
#    B      C        D
#0  11  111.0   6105.0
#1  22  222.0   9768.0
#2  33    NaN  13431.0
#3  44    NaN  17094.0
#4  55    NaN  20757.0
#5  66    NaN  24420.0
#6  77    NaN  28083.0
#7  88    NaN  31746.0
#8  99    NaN      NaN

Comments

[sham0909]

試してみたらできました。
ありがとうございました。