Python pandas 根據另一列的條件填充缺失值（NaN）

Question

我已經弄清楚如何使用df.fillna(method='ffill')用前一個單元格填充 NaN 值。

但是，我不確定如何將其基於以下條件：如果國家/地區名稱與其前一個單元格中的國家/地區名稱不同，則總大小寫單元格值應為 0，否則將NaN替換為前一個單元格的總大小寫值。

Answer 1

只需將groupby與fillna一起使用即可獲得所需的結果。 這里的columns是您要應用缺失值邏輯的所有列。

columns = ['total_cases', 'total_deaths', ...]
df[columns] = df.groupby('location')[columns].fillna(method='ffill').fillna(0)

請注意，您需要應用fillna兩次，一次使用正向填充，一次使用常量 0 來填充所有 nan 值。 這是為了確保在新組中開始的任何 nan 都用 0 填充。