Name Item Val1 Val2
x i 1 4
x j 2 5
y i 3 6
y j 0 6
所需要的功能:
令a为Val1的值,其中Name = x和Item = j然后是a = 2。
令b为Val2的值,其中Name = z和Item = j然后b = 0(如果它不存在)。
我试图将Name和Item设置为索引,然后使用df.loc[(‘x’, ‘j’), ‘Val1’],但这给了我2个问题:
它不返回单个值,并且如果该值不存在,则不给我0。
最佳答案
用try…except编写自己的函数
#df = df.set_index(['Name', 'Item'])
def yourlookup(Name, Item, Var):
try :
return df.loc[(Name, Item), Var]
except:
return 0
yourlookup('x', 'i', 'Val1') # 1
yourlookup('x', 'z', 'Val1') # 0