如果数据框中存在键，则将值读入字典

Question

如果数据字典中的某个键存在于数据框中，那么如何从该数据框中的某一列读取值。

d_shell = {
        'var' : 0,
        'var2' : 0,
        'var3' : 0,
        'var4' : 0,
        'var5' : 0
    }


for t in df['col']:
    new_dict = d_shell.fromkeys(['var', 'var2', 'var3', 'var4', 'var5'])
    for key in new_dict.items():
        if key in df['col']:
            value = df['other_col']
        else:
            value = 0

Answer 1

列表/字典理解是一件好事。

{k:df[df['col']==k]['other_col'].values for k,v in d_shell.items() if k in pd.unique(df['col']).tolist()}

您可以在PEP 274 -- Dict Comprehensions上找到更多信息。如果您是Python的新手，那么掌握这些技能之一就不那么容易了，但是一旦熟悉了它，就会上瘾。