增加重命名

tensnsorflow/ffm.py

@@ -21,11 +21,13 @@ def get_data():
     db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_prod')
     sql = "select * from esmm_data where stat_date >= '2018-11-20'"
     esmm = con_sql(db,sql)
+    esmm = esmm.rename(columns={1: "device_id"})
     print("esmm data ok")
     print(esmm.head())
     db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='eagle')
     sql = "select * from home_tab_click"
     temp = con_sql(db,sql)
+    temp = temp.rename(columns={0: "device_id"})
     print("click data ok")
     print(temp.head())
     df = pd.merge(esmm,temp,on = "device_id")
@@ -64,7 +66,9 @@ def transform(df,table):
     pd.io.sql.to_sql(df, table, yconnect, schema='jerry_test', if_exists='append', index=False)
     print("insert done")
 
+
 class multiFFMFormatPandas:
+
     def __init__(self):
         self.field_index_ = None
         self.feature_index_ = None
@@ -182,6 +186,7 @@ class multiFFMFormatPandas:
         else:
             return False
 
+
 if __name__ == "__main__":
     get_data()