#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# dessin de la synthèse des ages d'enfouissement des Grands Causses

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

#lecture des donnees
data = pd.read_csv('releves_TD2.csv') 
nom_optiq = ['pt_otiq1.','pt_otiq5','pt_otiq6']
nom_gnss = ['pt_gnss1.','pt_gnss5.','pt_gnss6.']

#creation des matrices vides
moy_optiq = np.zeros((3,3))
moy_gnss  = np.zeros((3,3))

#station totale
for count, i in enumerate(nom_optiq):
   print (i)
   moy_optiq[count,0] = np.round(np.mean(data.X[data['station'].str.contains(i)]),3)  # moyenne en X
   moy_optiq[count,1] = np.round(np.mean(data.Y[data['station'].str.contains(i)]),3)  # moyenne en Y
   moy_optiq[count,2] = np.round(np.mean(data.Z[data['station'].str.contains(i)]),3)  # moyenne en Z

for count, i in enumerate(nom_gnss):
   print (i)
   moy_gnss[count,0] = np.round(np.mean(data.X[data['station'].str.contains(i)]),3)  # moyenne en X
   moy_gnss[count,1] = np.round(np.mean(data.Y[data['station'].str.contains(i)]),3)  # moyenne en Y
   moy_gnss[count,2] = np.round(np.mean(data.Z[data['station'].str.contains(i)]),3)  # moyenne en Z

# exemple de difference
print(moy_optiq[0,:]-moy_optiq[1,:])