save project

func.py

 def yhdhg():
 def yhdhg():
     total = []
     total = []
     while True:
     while True:
-        unit= input("请输入：")
-        if unit== 'q':
+        w= input("请输入：")
+        if w== 'q':
             break
             break
         else:
         else:
-            total.append(unit)
+            try:
+                w=int(w)
+            except:
+                print('请重新输入  ')
+            else:
+                total.append(w)
     print(total)
     print(total)
 yhdhg()
 yhdhg()
\ No newline at end of file