匹配熊猫数据框和列表

时间:2021-05-19 13:46:19

标签: python pandas list dataframe matching

我有一个数据框 df 如下

 Number   PT
     
    5        AA
    
    64       BB
  
    7        CC

然后是另一个对象列表,

myList = [{'label': 'AA', 'value': 'AA', 'group': 'A'}, {'label': 'BB', 'value': 'BB', 'group': 'B'}]

我希望每个 PT 从列表中都有关联的组(如果可用),所以结果应该是这样的

    Number       PT    group
         
        5        AA    A
        
        64        BB    B
      
        7        CC    NOT_MATCHED
        

2 个答案:

答案 0 :(得分:0)

d = {'Number': [5, 64, 7], 'PT': ["AA", "BB", "CC"]}
df = pd.DataFrame(data=d)

myList = [{'label': 'AA', 'value': 'AA', 'group': 'A'}, {'label': 'BB', 'value': 'BB', 'group': 'B'}]    

for i, row in df.iterrows():
  for item in myList:
    if item['value'] == df['PT'][i]:
      df.at[i,'Group'] = item['group']
      break
    else:
      df.at[i,'Group'] = "NOT_MATCHED"

答案 1 :(得分:0)

尝试:

df['group'] = df.PT.map({tuple(i.values())[0]: tuple(i.values())[
                        2] for i in myList}).fillna('Not Matched')