python - 替换与另一列 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)