change eda

tensnsorflow/eda.py

@@ -5,18 +5,13 @@ import numpy as np
 import datetime
 import time
 
-def con_sql(db,sql):
+
+def con_sql(db, sql):
     cursor = db.cursor()
-    try:
-        cursor.execute(sql)
-        result = cursor.fetchall()
-        df = pd.DataFrame(list(result))
-    except Exception:
-        print("发生异常", Exception)
-        df = pd.DataFrame()
-    finally:
-        db.close()
-    return df
+    cursor.execute(sql)
+    result = cursor.fetchone()[0]
+
+    return result
 
 
 def test(days):
@@ -25,11 +20,11 @@ def test(days):
     sql = "select (select count(*) from train_data where stat_date = '{}' and y = 0)/(select count(*) " \
           "from train_data where stat_date = '{}' and z = 1)".format(start)
     db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
-    exp = con_sql(db, sql)[0].values.tolist()[0]
+    exp = con_sql(db, sql)
+    print(exp)
     sql = "select (select count(*) from train_data where stat_date = '{}' and y = 1 and z = 0)/(select count(*) " \
           "from train_data where stat_date = '{}' and z = 1)".format(start)
-    db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
-    click = con_sql(db, sql)[0].values.tolist()[0]
+    click = con_sql(db, sql)
     return start,exp,click