創建新的數據框列，保留另一列的第一個值

Question

我有一個包含兩列的數據框

import pandas as pd

df = pd.DataFrame()
df['A'] = [0.1, 0.1, 0.1, 0.1, 0.1]
df['B'] = [1.66, 1.66, 1.66, 1.66, 1.66]

  A    B
0.1 1.66
0.1 1.66
0.1 1.66
0.1 1.66
0.1 1.66

我想創建一個新列，在其中保存A列的第一個值，然后完成其余列值，如下所示：

  A     B          C
0.1  1.66       A[0]
0.1  1.66  B[0]*C[0]
0.1  1.66  B[1]*C[1]
0.1  1.66  B[2]*C[2]
0.1  1.66  B[3]*C[3] 

保持這種狀態

   A   B       C
1.66 0.1     0.1
1.66 0.1   0.166
1.66 0.1 0.27556
1.66 0.1 0.45743
1.66 0.1 0.75933

有什么辦法可以做到這一點？ ... 提前致謝！

Answer 1

嘗試這個：

df["C"] = df["B"].shift().fillna(df.loc[0, "A"]).cumprod()

Answer 2

如果您沒有大的df，這將起作用

data = {'A' : [0.1, 0.1, 0.1, 0.1, 0.1],
'B' : [1.66, 1.66, 1.66, 1.66, 1.66]}
df = pd.DataFrame(data)
df.at[0, 'C'] = df.iloc[0]['A']
for i in range(1, len(df)):
    df.at[i, 'C'] = df.iloc[i-1]['B'] * df.iloc[i-1]['C']
df