change main method for test

local/precitDIaryLocal.py → local/precitDiaryLocal.py

@@ -21,9 +21,13 @@ def test_con_sql(device_id):
     if not df.empty:
         df = df.rename(columns={0: "native_queue", 1: "nearby_queue", 2: "nation_queue", 3: "megacity_queue"})
         native_queue = df.loc[0, "native_queue"].split(",")
+        native_queue = [i.strip() for i in native_queue]
         nearby_queue = df.loc[0, "nearby_queue"].split(",")
+        nearby_queue = [i.strip() for i in nearby_queue]
         nation_queue = df.loc[0, "nation_queue"].split(",")
+        nation_queue = [i.strip() for i in nation_queue]
         megacity_queue = df.loc[0, "megacity_queue"].split(",")
+        megacity_queue = [i.strip() for i in megacity_queue]
         db.close()
         print(native_queue)
         print(nearby_queue)
@@ -48,7 +52,7 @@ def feature_en(x_list, device_id):
     data["minute"] = data["minute"].astype("category")
     # 虽然预测y，但ffm转化需要y，并不影响预测结果
     data["y"] = 0
-    data.to_csv("/Users/mac/utils/result/data.csv")
+    data.to_csv("/Users/mac/utils/result/data.csv",index=False)
     print(data)
     return data