graph finished and captor choice possible

Py-Script/courbe.py

@@ -1,5 +1,6 @@
+from pickletools import long1
 from secrets import choice
-from xmlrpc.client import boolean
+from turtle import goto
 import numpy as np
 import matplotlib.pyplot as plt
 import csv
@@ -8,16 +9,49 @@ import csv
 #! /usr/bin/env python3
 # -*- coding: UTF-8 -*-
     
-def function():
+    # x = tab[:,0]
-    #tab = np.zeros([1,9])
+    # y = tab[:,1]
+    # print(tab)
+    # return x , y
 
+#fonction choix du capteur à afficher : 
+#bool true si capteur veux être utiliser sinon false (tableau)
+def function():
     while True:
-        choice = input("Choisis un fichier csv de 1 à 1862: \n")
+        print("\n")
+        choice = input("(1) Choisis un fichier csv de 1 à 1862: ")
         print("\n")
         
         if float(choice) in range(1, 1863):
             rawdata = '../RawDataFiles/RawData' + choice + '.csv'   #Définit le fichier à ouvrir
 
+            choice2 = input("(2) Choisir un capteur à afficher en particulier dans le tableau? oui/non \n")
+            print("\n")
+
+            if choice2 == "oui":
+                while True:
+                    ccapteur = int(input("(3) Choisis parmis les capteurs (1, 2, 3, 4, 5, 6, 7, 8): "))
+                    print("\n")
+
+                    if ccapteur in (1,2,3,4,5,6,7,8):   #Affiche le capteur sélectionner
+                        with open(rawdata, newline='') as csvfile:
+                            reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC)
+                            x = [row for row in reader]
+                            int_x = np.array(x, int)
+                            column_index = [ccapteur + 1]
+
+                            for col in column_index:
+                                column_values = np.empty(0)
+                                for array in range(len(int_x)):
+                                    column_values = np.append(column_values, np.array(int_x[array][col-1]))
+                                plt.plot(column_values)
+                        plt.show()
+                        exit()
+
+                    elif ccapteur != (1,2,3,4,5,6,7,8):
+                        print("{} n'est pas un nombre valide.\n".format(int(ccapteur)))
+
+            elif choice2 == "non":  #Ouvre le csv file et affiche tout les capteurs
                 with open(rawdata, newline='') as csvfile:
                     reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC)
                     x = [row for row in reader]
@@ -35,13 +69,6 @@ def function():
         elif choice > '1862' or choice < '1':
             print("{} n'est pas un nombre valide.\n".format(choice))
 
-    # x = tab[:,0]
-    # y = tab[:,1]
-    # print(tab)
-    # return x , y
-
-bool 
-
 #Fonction d'affichage du graphique
 def printFig(axe_x , axe_y , title = "new plot" , xlabel = "Temps" , ylabel = "Tension"):
     plt.plot(axe_x , axe_y)
@@ -53,6 +80,7 @@ def printFig(axe_x , axe_y , title = "new plot" , xlabel = "Temps" , ylabel = "T
 
 #print(x)
 #print("y = " + str(y))
+
 def final():
     function()
     #printFig()