writeCSV (#326)

stripe · Mar 5, 2019 · 6751d8e · 6751d8e
1 parent 5c1b39c
commit 6751d8e
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/rainier-core/src/main/scala/com/stripe/rainier/repl/package.scala b/rainier-core/src/main/scala/com/stripe/rainier/repl/package.scala
@@ -1,6 +1,7 @@
 package com.stripe.rainier
 
 import com.stripe.rainier.sampler._
+import java.io._
 
 package object repl {
   def plot1D[N](seq: Seq[N])(implicit num: Numeric[N]): Unit = {
@@ -15,5 +16,20 @@ package object repl {
         .mkString("\n"))
   }
 
+  def writeCSV(path: String, seq: Seq[Map[String, Double]]): Unit = {
+    val fieldNames = seq.map(_.keys.toSet).reduce(_ ++ _).toList
+    val pw = new PrintWriter(new File(path))
+    pw.write(fieldNames.mkString(","))
+    seq.foreach { row =>
+      pw.write("\n")
+      fieldNames.tail.foreach { f =>
+        pw.write(row.get(f).map(_.toString).getOrElse(""))
+        pw.write(",")
+      }
+      pw.write(row.get(fieldNames.head).map(_.toString).getOrElse(""))
+    }
+    pw.close
+  }
+
   implicit val rng: RNG = RNG.default
 }