поменяйте панд df на python dict

Question

Я хотел бы изменить следующий объект DataFrame pandas на словарь 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 и 2 объекта, которые я должен получить из DataFrame

(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

Как мне сделать .query для работы с python dict?

Answer 1

pandas.DataFrame.to_dict по умолчанию отображает данные в виде диктов.

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

Если вы хотите, чтобы они были списками списков:

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