用與下一列對應的顏色填充 nan 值

Question

我想用與該條件對應的顏色類型填充 nan 值。 例如，如果它是棕色且條件具有 nan 值，則 nan 應填充 2.0數據集圖像

塊引用

condition   color_type  length(m)   height(cm)  X1  X2  duration
0   2.0 Brown Tabby 0.80    7.78    13  9   73
1   1.0 White   0.72    14.19   13  9   1862
2   NaN Brown   0.15    40.90   15  4   752
3   1.0 White   0.62    17.82   0   1   755
4   2.0 Black   0.50    11.06   18  4   52

Answer 1

您可以使用 boolean 掩碼：

cond = df['color_type'] == 'Brown'
df.loc[cond, "condition"] = df.loc[cond, "condition"].fillna(2.0)

---

對於許多 colors，您可以這樣做：

cond = df['color_type'].isin(your_list_of_colors)

Answer 2

你可以這樣做 df["condition"]=df["condition"].fillna(df.groupby("color_type")["condition"].transform(lambda x:x.mode()[0]))這應該可以解決您的問題