python - 在字典中访问 Pandas 面具

Question

我有一个字典，其中包含多个 pandas 掩码作为特定数据帧的字符串，但我找不到使用这些掩码的方法。

这是一个简短的可重现示例:

df = pd.DataFrame({'age' : [10, 24, 35, 67], 'strength' : [0 , 3, 9, 4]})

masks = {'old_strong' : "(df['age'] >18) & (df['strength'] >5)",
        'young_weak' : "(df['age'] <18) & (df['strength'] <5)"}

我想做类似的事情:

df[masks['young_weak']]

但是因为掩码是一个字符串，所以我得到了错误

KeyError: "(df['age'] <18) & (df['strength] <5)"

Answer 1

使用DataFrame.query改变字典:

masks = {'old_strong' : "(age >18) & (strength >5)",
        'young_weak' : "(age <18) & (strength <5)"}

print (df.query(masks['young_weak']))
   age  strength
0   10         0