from pyeconlab import DynamicProductLevelExportSystem
#-Local Imports-#
from dataset_info import TARGET_DATASET_DIR, CHAPTER_RESULTS
DATASET_DIR = TARGET_DATASET_DIR['nber']
RESULTS_DIR = CHAPTER_RESULTS["G"]
data = pd.read_hdf(DATASET_DIR + "nber-export-sitcr2l4-1962to2000.h5", "D")
data = data.rename(columns={
'eiso3c': 'country',
'sitc4': 'productcode',
'value': 'export'
})
data = data.set_index(["year"])
system = DynamicProductLevelExportSystem()
system.from_df(data)
#-Year 2000-#
ys = system[2000]
ys.rca_matrix(complete_data=True)
ys.mcp_matrix()
ys.compute_pci()
ys.auto_adjust_pci_sign()
pci = ys.pci.copy()
#-Example Proximity Values-#
from pyeconlab.trade.classification import SITCR2
sitc_to_name = SITCR2().code_description_dict()
prox1 = ys.proximity_matrix()
products1 = ["8423", "0711"]
