plot still appearing even if number not valid fixed

Py-Script/courbe.py

@@ -1,19 +1,20 @@
-import math
 from secrets import choice
+from tracemalloc import stop
 import numpy as np
 import matplotlib.pyplot as plt
-import struct
 import csv
 
 #"C:\Users\quent\OneDrive\Bureau\ENSC\TransD\Framboisier\02400001.TXT"
 #! /usr/bin/env python3
 # -*- coding: UTF-8 -*-
 
-tab = np.zeros([1,9])
+def function():
+    tab = np.zeros([1,9])
 
-#def tab_choix():
-while True:
+    #def tab_choix():
+    while True:
         choice = input("Choisis un fichier csv de 1 à 1862: \n")
+        print("\n")
         
         if float(choice) in range(1, 1863):
             rawdata = '../RawDataFiles/RawData' + choice + '.csv'
@@ -26,11 +27,13 @@ while True:
             break
         
         elif choice > '1862' or choice < '1':
-        print("{} n'est pas un nombre valide.".format(choice))
-        break
+            print("{} n'est pas un nombre valide.\n".format(choice))
+            exit()
     
-x = tab[:,0]
-y = tab[:,1]
+    x = tab[:,0]
+    y = tab[:,1]
+    print(tab)
+    return x , y
 
 #Fonction d'affichage du graphique
 def printFig(axe_x , axe_y , title = "new plot" , xlabel = "Temps" , ylabel = "Tension"):
@@ -41,7 +44,10 @@ def printFig(axe_x , axe_y , title = "new plot" , xlabel = "Temps" , ylabel = "T
     plt.show()
     return 0
 
-print(tab)
 #print(x)
 #print("y = " + str(y))
-printFig(x,y)
+def final():
+    x, y = function()
+    printFig(x,y)
+
+final()