Pandas，按组向后填充

Question

我有这个问题 dataframe 下面的代码

import numpy as np
import pandas as pd
from numpy import nan
tostk = np.asarray([['A', nan, 6.0, nan, nan],
       ['A', 3.0, nan, nan, nan],
       ['A', nan, nan, 9.0, nan],
       ['A', nan, 5.0, nan, nan],
       ['A', nan, nan, nan, 7.0],
       ['B', nan, 8.0, nan, 7.0],
       ['B', nan, nan, 6.0, nan],
       ['B', 6.0, nan, nan, 8.0],
       ['B', 5.0, nan, nan, 6.0],
       ['B', nan, nan, 4.0, nan]])
pd.DataFrame(tostk)

我需要用第一个值替换每个类别（A 和 B）的 nan 值所以我尝试了 bfill 但“bfill”的问题是如果该值属于 B 类，它将填充 A 类中的值

预期结果

res = np.asarray([['A', 3.0, 6.0, 9.0, 7.0],
           ['A', 3.0, 5.0, 9.0, 7.0],
           ['A', nan, 5.0, 9.0, 7.0],
           ['A', nan, 5.0, nan, 7.0],
           ['A', nan, nan, nan, 7.0],
           ['B', 6.0, 8.0, 6.0, 7.0],
           ['B', 6.0, nan, 6.0, 8.0],
           ['B', 6.0, nan, 4.0, 8.0],
           ['B', 5.0, nan, 4.0, 6.0],
           ['B', nan, nan, 4.0, nan]])
    pd.DataFrame(res)

欢迎任何想法