在Python中基于列表和字典编写多列代码

Question

我有以下熊猫数据帧

 OfferPreference_A  OfferPreference_B   OfferPreference_C   
            A          B                  A                  
            B          C                  C                 
            C          S                  G  

我在所有列下都有下面的唯一值字典

  dict1={A:1, B:2, C:3, S:4, G:5, D:6}

我还有一份专栏作家的名单

  columnlist=['OfferPreference_A', 'OfferPreference_B', 'OfferPreference_C']

我试图得到下表作为输出

    OfferPreference_A   OfferPreference_B   OfferPreference_C   
               1           2                  1                  
               2           3                 3                  
               3           4                  5  

我该怎么做

Answer 1

您可以使用map来实现如下所示的效果，假设值总是匹配的

for col in columnlist:
    df[col] = df[col].map(dict1)