替換與另一列 pandas 中的特定值相對應的列中的空值

Question

我有一個如下的數據框：

import pandas as pd
df = pd.DataFrame({'Country': ['USA','USA','MEX','IND','UK','UK','UK'],
                   'Region': ['Americas','NaN','NaN','Asia','Europe','NaN','NaN'],
                   'Flower': ['Rose','Lily','Lily','Orchid','Petunia','Lotus','Dandelion']})

我想用其他區域的值替換 Region 中的NaN值。 我的意思是，如果國家是美國或墨西哥，地區應該是美洲，如果國家是英國，地區應該是歐洲。

預期的輸出是

result = pd.DataFrame({'Country': ['USA','USA','MEX','IND','UK','UK','UK'],
                       'Region': ['Americas','Americas','Americas','Asia','Europe','Europe','Europe'],
                       'Flower': ['Rose','Lily','Lily','Orchid','Petunia','Lotus','Dandelion']})

我想知道一種簡單的方法，而不必編寫 if else 語句。

Answer 1

如果 ffill 不是一個選項，您可以嘗試以下代碼，

df['Region'] = np.select((df.Country.isin(['USA', 'MEX']), df.Country == 'UK'),
                         ('Americas', 'Europe'), df.Region)