import numpy as np
import pandas as pd

from Census.commons import AdjMatrix, load_csv, get_dataset_by_area

census = './data/cens_y.csv'
Areas, X = load_csv(census, use_yhat=True)
data = get_dataset_by_area(Areas, X)

areas = [a for a, *_ in data]

print(f'Area codes={areas}')

A = AdjMatrix('./data/matrice_adiacenza.csv')
print(A.adjacent(45, 46))
print(A.get_adjacent(50))