import numpy as np
import pylab as py
import matplotlib.pyplot as plt



h = np.array([10, 50,90])              # différence de hauteur en cm
t = np.array([ 138, 30, 20])              # temps mis pour remplir le verre en s
v=60                                    # Volume du verre en mL
Q=v/t

xh=np.linspace(0,100,1000)

u = np.array([6.26 , 10.9, 14.8, 22.6])           # tension aux bornes de la résistance en V
i = np.array([2.0,3.3 , 4.8, 7.1 ])          # intensité traversant le circuit en A
xu=np.linspace(0,30,1000)

plt.figure(1)
plt.subplot(121)
ph=np.polyfit(h,Q,1)                        # régression linéaire de Q
plt.plot(h,Q, 'rs', xh,ph[0]*xh+ph[1],'-')  # on trace les données et la régression linéaire


plt.xlabel(r'$\Delta h \, {\rm (cm)}$  ') # on n'oublie pas de nommer les axes...
plt.ylabel(r'$Q \,{\rm (mL.s^{-1})}$  ')  # ....avec les unités !! 
plt.axis([0, 1.1*max(h), 0, 1.1*max(Q)]) # on fixe les limites des axes



# on fait la même chose pour la caractéristique de la résistance électrique
plt.subplot(122)

pu=np.polyfit(u,i,1)
plt.plot(u,i, 'bo', xu,pu[0]*xu+pu[1],'k-')



plt.xlabel(r'$U \, {\rm (V)}$  ')
plt.ylabel(r'$I \,{ \rm (  mA)}$  ')
plt.axis([0, 1.1*max(u), 0, 1.1*max(i)])
 
py.show() # on affiche le graphe