change test file

tensnsorflow/multi.py

@@ -199,12 +199,13 @@ def con_sql(db,sql):
 
 def test():
     sql = "select stat_date,cid_id from esmm_train_data e where stat_date = '{}' limit 60".format("2019-04-25")
-    df = spark.createDataFrame(spark.sql(sql).rdd.map(lambda x:(x[0],x[1])).zipWithIndex()
-                               .map(lambda x:(x[1],x[0][0],x[0][1]))).toDF("ind","k","v")
-    df.show(6)
+
+    df = spark.sql(sql)
 
     # df.write.csv('/recommend/tr', mode='overwrite', header=True)
-    df.write.format("avro").save("/recommend/tr/avro.csv")
+    df.select("stat_date", "cid_id").write.format("avro").save("/recommend/tr/avro.csv")
+    # df.write.format("avro").save("/recommend/tr/avro.csv")
+
 
 
     #