feat(script): Move big code chunks out of script

01-codechunks/_prep-data.py

@@ -0,0 +1,37 @@
+from pathlib import Path
+import re
+## standard imports
+from IPython.core.display import Markdown as md
+import numpy as np
+import pandas as pd
+from matplotlib import pyplot as plt
+import seaborn as sns
+from tabulate import tabulate
+import bibtexparser
+
+sns.set_style("whitegrid")
+
+DATA_DIR=Path("./02-data")
+RAW_DATA=DATA_DIR.joinpath("raw")
+WORKING_DATA=DATA_DIR.joinpath("intermediate")
+PROCESSED_DATA=DATA_DIR.joinpath("processed")
+SUPPLEMENTARY_DATA=DATA_DIR.joinpath("supplementary")
+
+from src import prep_data
+
+# raw database-search results
+bib_sample_raw_db = prep_data.bib_library_from_dir(RAW_DATA)
+# the complete library of sampled (and working) literature
+bib_sample = prep_data.bib_library_from_dir(WORKING_DATA)
+
+# load relevant studies
+from src import load_data
+
+bib_df = prep_data.observations_with_metadata_df(
+    raw_observations = load_data.from_yml(PROCESSED_DATA),
+    study_metadata = prep_data.bib_metadata_df(bib_sample),
+    country_groups = prep_data.country_groups_df(Path(f"{SUPPLEMENTARY_DATA}/wb-country-groupings.xlsx")),
+)
+raw_observations = None
+zot_df = None
+df_country_groups = None