# ### Read Data # + # ######################################################### df_jobs_anal = get_df_jobs_anal() df_jobs_anal_i = df_jobs_anal # ######################################################### df_atoms_sorted_ind = get_df_atoms_sorted_ind() # ######################################################### magmom_data_dict = read_magmom_comp_data() # ######################################################### df_magmoms = get_df_magmoms() # ######################################################### df_jobs_paths = get_df_jobs_paths() # ######################################################### df_magmoms = df_magmoms.set_index("job_id") # ######################################################### df_jobs_oh_anal = get_df_jobs_oh_anal() # + df_ind = df_jobs_anal_i.index.to_frame() df_jobs_anal_i = df_jobs_anal_i.loc[df_ind[df_ind.job_type == "oer_adsorbate"].index] df_jobs_anal_i = df_jobs_anal_i.droplevel(level=0)
# Pickling data ########################################### import os; import pickle directory = os.path.join( os.environ["PROJ_irox_oer"], "dft_workflow/job_analysis/compare_magmoms", "out_data") if not os.path.exists(directory): os.makedirs(directory) path_i = os.path.join(directory, "df_magmoms.pickle") with open(path_i, "wb") as fle: pickle.dump(df_magmoms, fle) # ######################################################### # + from methods import get_df_magmoms df_magmoms_tmp = get_df_magmoms() # - # ######################################################### print(20 * "# # ") print("All done!") print("Run time:", np.round((time.time() - ti) / 60, 3), "min") print("analyse_jobs.ipynb") print(20 * "# # ") # ######################################################### # + active="" # # #