时间戳

首页 > 解决方案 > 替换与另一列熊猫中的特定值相对应的列中的空值

python - 替换与另一列熊猫中的特定值相对应的列中的空值

问题描述

我有一个数据框如下：

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']})

我想将NaNRegion 中的值替换为其他区域的值。我的意思是，如果该国家/地区是美国或墨西哥，则该地区应该是美洲，如果该国家/地区是英国，则该地区应该是欧洲。

预期的输出是

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 语句。

标签： pythonpandas

解决方案

如果 ffill 不是一个选项，您可以尝试下面的代码，

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

推荐阅读