如何在pandas中映射列值？

Question

在“我的文件”中，有一列包含不同的等级（columns name='grades）。你知道吗

例如：9150K、9260K、柴油机、ADBlU等。。你知道吗

对于所有这些等级，我需要把它们分为几个等级

例如: 91 50K= Petrol

在我的python中，我如何做到这一点？注意，我可以将整个列传递给函数。但是函数必须用正确的值重写每一行的值

def checkgrades(data):
df['Grades']=???
???

按照下面的答案我试着

 df_dips=map_dips_grades(df_dips)
df_sales = df_sales.merge(df_dips, left_on=['Site Name', 'Date','GradeNo'],
                              right_on=['Site', 'Dip Time', 'Product'], how='left').fillna(0)



def map_dips_grades(data):

    d1 = {'Diesel': ['DIESEL', 'DIESEL 1'],
          'Unleaded': ['91','91 UNLEADED'],
          'PULP':['95','95 ULP'],
          'PULP98':['98','98 20K'],
          'Vortex Diesel':['DIESEL ULT R'],
          'Adblue':['ADBLU','ADO']}

    d = {k: oldk for oldk, oldv in d1.items() for k in oldv}

    data['Product'].map(d)
    return data

但我得到了

ValueError: You are trying to merge on int64 and object columns. If you wish to proceed you should use pd.concat

Answer 1

您可以在Grades和^{}中创建所有可能值的字典：

#test all possible unique values
print (df['Grades'].unique())

d = {'91 50K':'Petrol','92 60K':'Petrol','DIESEL':'Diesel',...}

df['Grades'] = df['Grades'].map(d)

另一个可能减少键入的字典是dict of list：

d1 = {'Petrol':['91 50K','92 60K'],
      'Diesel':['DIESEL']}

#swap key values in dict
#http://stackoverflow.com/a/31674731/2901002
d = {k: oldk for oldk, oldv in d1.items() for k in oldv}
print (d)
{'91 50K': 'Petrol', '92 60K': 'Petrol', 'DIESEL': 'Diesel'}

df['Grades'] = df['Grades'].map(d)