根据字典中给定的条件将数据插入新列

Question

如果满足条件，我有字典指定行应采用的值

dict_map = {
    "Yes" : {'Sex':'F', 'Pregnant': 'Y'},
    "No" : {'Sex':'F', 'Pregnant': 'N'},
    "N/A" : {'Sex': 'M'},
    }

数据框和列表

df = pd.DataFrame(
{
    'ID': ['AB01', 'AB02', 'AB03', 'AB04', 'AB05','AB06'],
    'Sex': ["F","M","M",np.nan,"F","F"],
    'Pregnant': ['Y', 'N',np.nan, 'Y', '', 'N']
    
 }
 )

 cols_of_interest = ["Sex","Pregnant"]

最终的数据框应如下所示，其中 Status 是新列。不使用合并功能如何做到这一点？

Answer 1

您可以创建数据框然后merge

out = df.merge(pd.DataFrame(dict_map).T.reset_index(),how='left')
Out[139]: 
     ID  Sex Pregnant index
0  AB01    F        Y   Yes
1  AB02    M        N   NaN
2  AB03    M      NaN   N/A
3  AB04  NaN        Y   NaN
4  AB05    F            NaN
5  AB06    F        N    No