将df更改为python di

Question

我想将下面的DataFrame对象更改为python字典，然后用energy_emob_BB和energy_emob_BE获取值。你知道吗

emob_energy = pd.DataFrame(session.query(abbb_emob.id,
                                         abbb_emob.scenario,
                                         abbb_emob.region,
                                         abbb_emob.energy).filter(
                           abbb_emob.scenario == 'ES2030').all())

energy_emob_BB = float(emob_energy.query('region=="BB"')['energy'])
energy_emob_BE = float(emob_energy.query('region=="BE"')['energy'])

这里是DataFrame的一些输出和我必须从DataFrame获得的2个对象

(Pdb) emob_energy
   id scenario region     energy
0   1   ES2030     BB  2183000.0
1   2   ES2030     BE  1298333.0

(Pdb) energy_emob_BB
2183000.0

(Pdb) energy_emob_BE
1298333.0

如何使用python dict生成.query？你知道吗

Answer 1

pandas.DataFrame.to_dict默认情况下，在dicts中呈现dicts中的数据。你知道吗

df_dict = emob_energy.to_dict()
print(df_dict['energy_emob_bb']) # {0: 2183000.0, 1: 1298333.0, ...}

如果您想将它们作为列表的dict：

df_dict = emob_energy.to_dict(orient='list')