如何从另一列及以上行中的值填充 pandas dataframe 中的 nan 值？

Question

我有 df 如下：

df = pd.DataFrame({"A":[0,np.nan,0,0,np.nan,1,np.nan,1,0,np.nan],
                   "B":[0,1,0,0,1,1,1,0,0,0]})

现在，我需要将 A 列中的 nan 值替换为 B 列和上一行的值。 例如：A 列的第 2 行应为 0，第 7 行应为 1，依此类推。

我定义了这个 function 但它无法尝试应用于 dataframe

def impute_with_previous_B(df):
    for x in range(len(df)):
        if pd.isnull(df.loc[x,"A"]) == True:
            df.loc[x,"A"] = df.loc[x-1,"B"]

df["A"] = df.apply(lambda x: impute_with_previous_B(x),axis=1)

你能告诉我那个function有什么问题吗？

Answer 1

df['A'] = df['A'].fillna(df['B'].shift())


     A  B
0  0.0  0
1  0.0  1
2  0.0  0
3  0.0  0
4  0.0  1
5  1.0  1
6  1.0  1
7  1.0  0
8  0.0  0
9  0.0  0