# -*- coding: utf-8 -*-
"""
Created on Fri Jul  8 14:38:50 2022

@author: starons
"""

"""
Représente graphiquement un nuage de points
"""

# On importe les bibliothèques utiles
# -----------------------------------
import numpy as np                                      # pour la gestion des tableaux
from matplotlib import pyplot as plt                    # pour les représentations graphiques

# On rentre à la main les données
# -------------------------------
Lambda=np.array([404.7,435.8,480.0,546.1,578.1,615])    # longueur d'onde en nm 
n=np.array([1.7761,1.7652,1.7475,1.7358,1.7294,1.7229]) # indice de réfraction

# Ou on les récupère dans un fichier .csv
# ---------------------------------------
import csv
with open("prisme.csv") as fichier :
    reader = csv.reader(fichier, delimiter = ",")
    entete = next(reader)                               # saute la première ligne
    entete = next(reader)                               # saute la deuxième ligne
    données = np.array(list(reader)).astype("float")    # stocke les données dans un tableau numpy
Lambda=données [:,0]                                    # toutes les lignes de la première colonne
n=données [:,1]                                         # toutes les lignes de la deuxième colonne 

# Représentation graphique
# ------------------------
plt.plot(Lambda,n,'+')
plt.xlabel(r"$\lambda(nm)$")                            # langage Latex entre les $ pour faire joli
plt.ylabel("n")
plt.title("Dispersion de la lumière par un prisme");plt.grid()