Explorar el Código

Write 3 different output files

theenglishway (time) hace 2 años
padre
commit
e45fe1ea76
Se han modificado 1 ficheros con 7 adiciones y 1 borrados
  1. 7 1
      src/main/scala/Main.scala

+ 7 - 1
src/main/scala/Main.scala

@@ -28,5 +28,11 @@ object Main extends App {
   println(stats.size)
 
   val analysis = GamesAnalysis(teams_output, games_output, stats_output)
-  GamesAnalysis.writeToCsv(analysis.merged, "output/merged_df")
+
+  Map(
+    "output/teams_games_df" -> analysis.teams_games.sort("game_id", "team_id"),
+    "output/stats_games_pts_df" -> analysis.stats_games_pts
+      .sort("game_id", "team_id"),
+    "output/final_df" -> analysis.merged.sort("game_id", "team_id")
+  ).foreachEntry((path, df) => GamesAnalysis.writeToCsv(df, path))
 }