from mpl_toolkits.basemap import Basemap
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split 
import pandas as p

file = p.read_csv('files/sigfox_dataset_antwerp.csv')
columns = file.columns
x = file[columns[0:84]]
y = file[columns[85:]]

random_state=42
x_train, x_test_val, y_train, y_test_val = train_test_split(x.values, y.values, test_size=0.3, random_state=random_state)
x_val, x_test, y_val, y_test = train_test_split(x_test_val, y_test_val, test_size=0.5, random_state=random_state)


map = Basemap(llcrnrlon=4.32,llcrnrlat=51.17,urcrnrlon=4.46,urcrnrlat=51.25, lat_0 = 51.2, lon_0 = 4.3)
map.arcgisimage(service='World_Topo_Map', xpixels = 10000, ypixels = 6000,dpi=600, verbose= True)

map.scatter(y_train[:,1],y_train[:,0],c='r',s=0.15,marker='.',alpha=0.9)
map.scatter(y_val[:,1],y_val[:,0],c='k',s=0.1,marker='s',alpha=0.9)
map.scatter(y_test[:,1],y_test[:,0],c='b',s=0.1,marker='s',alpha=0.9)


plt.savefig('figures/Antwerp_train_val_test.png', dpi=600, bbox_inches='tight')
plt.show()