feldtruby 0.3.10 → 0.3.11

data/lib/feldtruby/statistics.rb

@@ -156,17 +156,21 @@ end
 # Plotting data sets in R with ggplot2 and save them to files.
 module FeldtRuby::Statistics::Plotting
 
-  def plot_2dims(csvFilePath, graphFilePath, xName, yName, title = "scatterplot", width = 1200, height = 900)
+  def plot_2dims(csvFilePath, graphFilePath, xName, yName, title = "scatterplot", format = "pdf", width = 7, height = 5)
 
     include_library("ggplot2")
 
+    gfxFile = File.dirname(graphFilePath) + "/" + File.basename(graphFilePath, "." + format) + "." + format
+
     pre = [
-      "td <- read.csv(#{csvFilePath.inspect}",
-      "png(#{graphFilePath.inspect}, width=#{width}, height=#{height})"
+      "data <- read.csv(#{csvFilePath.inspect})",
+      "#{format}(#{gfxFile.inspect}, width=#{width}, height=#{height})"
     ]
 
-    plot = yield()
-    plot.last << " theme_bw(base_size = 12, base_family = \"\")"
+    #plot = ["suppressWarnings( " + yield().join(" ") + " + theme_bw(base_size = 12, base_family = \"\") )"]
+    plot = [yield().join(" ") + " + theme_bw(base_size = 12, base_family = \"\")"]
+    #plot = yield()
+    #plot << " + theme_bw(base_size = 16, base_family = \"\")"
 
     post = [
       "dev.off()"
@@ -178,10 +182,10 @@ module FeldtRuby::Statistics::Plotting
   end
 
   # Scatter plot of columns xName vs yName in csvFilePath is saved to graphFilePath.
-  def scatter_plot(csvFilePath, graphFilePath, xName, yName, title = "scatterplot", smoothFit = true, width = 1200, height = 900)
-    plot_2dims(csvFilePath, graphFilePath, xName, yName, title, width, height) {
+  def scatter_plot(csvFilePath, graphFilePath, xName, yName, title = "scatterplot", smoothFit = true, format = "pdf", width = 7, height = 5)
+    plot_2dims(csvFilePath, graphFilePath, xName, yName, title, format, width, height) {
       [
-        "ggplot(td, aes(#{xName}, #{yName})) + ",
+        "ggplot(data, aes(#{xName}, #{yName})) + ",
         "  geom_point(shape = 1) + ", # Each point is non-filled circle
         (smoothFit ? "  geom_smooth() + " : nil),
         "  ggtitle(#{title.inspect})"
@@ -190,9 +194,9 @@ module FeldtRuby::Statistics::Plotting
   end
 
   # Scatter plot of columns xName vs yName in csvFilePath is saved to graphFilePath.
-  def hexbin_heatmap(csvFilePath, graphFilePath, xName, yName, title = "heatmap", bins = 30, width = 1200, height = 900)
-    plot_2dims(csvFilePath, graphFilePath, xName, yName, title, width, height) {
-      [ "ggplot(td, aes(#{xName}, #{yName})) + geom_hex( bins = #{bins} ) + ggtitle(\"#{title}\")"]
+  def hexbin_heatmap(csvFilePath, graphFilePath, xName, yName, title = "heatmap", bins = 30, format = "pdf", width = 7, height = 5)
+    plot_2dims(csvFilePath, graphFilePath, xName, yName, title, format, width, height) {
+      [ "ggplot(data, aes(#{xName}, #{yName})) + geom_hex( bins = #{bins} ) + ggtitle(\"#{title}\")"]
     }
   end
 end

data/lib/feldtruby/version.rb

@@ -1,3 +1,3 @@
 module FeldtRuby
-  VERSION = "0.3.10"
+  VERSION = "0.3.11"
 end

data/test/test_statistics.rb

@@ -89,11 +89,43 @@ require 'feldtruby/minitest_extensions'
 describe "Test Statistics but with the extensions to MiniTest framework" do
   it "can use assert_same_proportions" do
     assert_similar_proportions( [1]*10 + [2]*10 )
-    # This should fail but I found now way to test it since it uses the MiniTest framework itself...
+    # This should fail but I found no way to test it since it uses the MiniTest framework itself...
     # assert_similar_proportions( [1]*60 + [2]*40 )
   end
 
   it "can use must_have_similar_proportions" do
     ([1]*10 + [2]*10).must_have_similar_proportions
   end
+end
+
+describe "Plotting" do
+
+  it "can do a scatter plot" do
+
+    d = File.dirname(__FILE__) + "/"
+    filename = d + "tmp.csv"
+    out = d + "scatterplot.pdf"
+    RC.scatter_plot(filename, out, "size", "height", 
+      "Scatterplot", true)
+
+    File.exist?(out).must_equal true
+
+    File.delete out
+
+  end
+
+  it "can do a hexbin heatmap plot" do
+
+    d = File.dirname(__FILE__) + "/"
+    filename = d + "tmp.csv"
+    out = d + "heatmap.pdf"
+
+    RC.hexbin_heatmap(filename, out, "size", "height", 
+      "Hexbin heatmap", 30)
+
+    File.exist?(out).must_equal true
+
+    File.delete out
+
+  end
 end

data/test/tmp.csv

@@ -0,0 +1,799 @@
+balance,height,size
+0.0,14,73
+0.0,2,2
+0.0,1,1
+0.0,6,7
+0.0,2,3
+0.0,1,1
+0.0,3,4
+0.0,1,1
+0.0,1,1
+0.0,2,3
+0.0,1,1
+0.0,3,3
+0.0,1,1
+0.0,6,12
+0.0,2,2
+0.3726779962499649,3,8
+0.0,5,7
+0.0,1,1
+0.0,2,3
+0.0,1,1
+0.0,1,1
+0.0,4,9
+0.0,1,1
+0.0,1,1
+11.027239001672177,30,364
+0.0,1,1
+1.0,4,5
+2.160246899469287,7,18
+0.0,1,1
+0.0,2,2
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.5,3,6
+0.0,1,1
+0.0,1,1
+0.0,2,3
+0.0,1,1
+0.5,3,7
+0.0,1,1
+0.0,1,1
+0.0,2,3
+1.0,5,13
+0.0,1,1
+2.598076211353316,9,35
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,5,6
+0.0,2,3
+0.0,1,1
+0.0,2,2
+0.0,2,2
+0.0,1,1
+12.387090053761618,34,419
+2.280350850198276,8,44
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+2.8,8,26
+0.0,1,1
+5.15363949069005,14,179
+0.0,2,3
+1.5,5,7
+2.5,8,18
+0.0,1,1
+0.0,1,1
+0.5,3,4
+0.0,1,1
+0.0,3,3
+0.0,1,1
+79.00158226263572,189,7555
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,3
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,3,3
+0.0,2,2
+0.0,3,3
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,4,4
+0.3726779962499649,3,11
+2.0,6,13
+0.0,1,1
+0.0,1,1
+3.5,9,27
+0.5,3,4
+0.0,8,25
+0.0,1,1
+0.0,1,1
+0.0,1,1
+29.272664533466862,66,1256
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,22,191
+0.0,1,1
+0.8660254037844386,4,7
+0.0,1,1
+1.4142135623730951,5,7
+7.22841614740048,19,56
+0.0,1,1
+8.993825042154695,22,104
+0.0,2,2
+0.0,1,1
+1.0,4,5
+0.0,2,2
+0.0,1,1
+1.632993161855452,6,12
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,4,5
+0.0,3,3
+0.0,1,1
+1.124858267715973,5,19
+0.9428090415820634,4,6
+0.0,4,5
+0.0,1,1
+0.0,1,1
+0.0,1,1
+1.8856180831641267,6,10
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,3,3
+0.0,3,3
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,3
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,4
+0.0,3,3
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+13.0,31,943
+0.0,3,3
+3.5,9,22
+0.0,1,1
+0.0,1,1
+9.5,24,83
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.4714045207910317,3,5
+0.0,3,3
+0.0,2,3
+0.0,1,1
+2.8284271247461903,8,18
+1.5,5,7
+23.338094752285727,52,396
+0.0,1,1
+0.0,1,1
+0.0,2,3
+22.37604746151563,54,482
+3.091206165165235,9,24
+0.0,2,3
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.5,3,4
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,2,2
+4.0,10,30
+6.800735254367722,22,286
+0.0,1,1
+8.48528137423857,20,262
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,7,8
+0.0,1,1
+0.0,2,2
+0.4714045207910317,4,8
+0.0,2,3
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,3,3
+0.0,1,1
+0.0,1,1
+0.6998542122237652,4,15
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,3,5
+0.0,1,1
+0.0,1,1
+2.4411439272335804,10,43
+0.0,2,2
+0.0,1,1
+0.0,2,2
+0.0,2,2
+0.0,1,1
+0.5,6,16
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+3.0,8,11
+0.0,3,3
+25.0,52,537
+0.0,1,1
+0.0,1,1
+0.0,2,2
+2.0,6,7
+0.8660254037844386,4,7
+0.0,1,1
+8.0,18,82
+0.0,1,1
+3.0,9,21
+1.247219128924647,5,15
+0.0,11,34
+0.0,2,2
+0.0,1,1
+0.0,3,5
+0.0,1,1
+0.0,3,5
+0.0,1,1
+0.0,1,1
+1.4142135623730951,5,7
+0.0,1,1
+0.0,1,1
+0.0,3,3
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,21,81
+0.0,3,3
+0.0,1,1
+0.0,1,1
+0.0,1,1
+1.5,12,58
+0.0,1,1
+0.0,1,1
+0.6998542122237652,4,13
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,5,8
+0.0,2,2
+1.0,5,9
+0.0,8,15
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+1.0,5,11
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,45,489
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,31,369
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.4714045207910317,3,6
+0.0,3,3
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,5,8
+0.5,3,4
+0.0,9,16
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,3,7
+0.0,3,3
+0.0,1,1
+0.0,3,3
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,4,5
+0.0,2,3
+0.0,1,1
+0.0,2,2
+0.0,1,1
+10.03327796219494,25,222
+0.0,1,1
+1.5,5,9
+0.0,2,2
+0.0,1,1
+0.0,2,2
+0.0,4,11
+0.0,1,1
+5.715476066494082,15,56
+0.9428090415820634,4,9
+0.0,1,1
+0.0,5,6
+0.0,1,1
+0.5,3,4
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.5,3,4
+0.0,1,1
+0.5,3,7
+5.338539126015656,15,102
+0.0,1,1
+0.0,17,39
+0.0,1,1
+0.5,3,4
+0.0,1,1
+0.0,5,7
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.816496580927726,4,8
+0.0,3,4
+0.0,1,1
+36.50342449688796,90,1720
+0.0,1,1
+0.0,4,6
+0.0,18,139
+0.0,1,1
+3.7416573867739413,14,114
+0.0,2,2
+0.0,3,7
+1.0,4,6
+0.0,2,2
+0.0,1,1
+4.714045207910317,12,48
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+2.943920288775949,9,22
+0.0,3,3
+1.5,5,13
+0.0,1,1
+0.0,8,43
+0.0,1,1
+0.0,1,1
+0.7483314773547883,4,15
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,5,6
+0.0,2,2
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.7284313590846836,4,12
+0.8660254037844386,4,8
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.5,3,4
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+28.994252303976847,64,1219
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,2,2
+0.0,1,1
+0.0,1,1
+1.0,5,9
+0.0,9,33
+0.5,4,6
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.4714045207910317,3,7
+0.0,9,14
+0.0,1,1
+1.0,5,16
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,5,6
+0.0,2,2
+0.0,1,1
+0.0,18,124
+0.0,2,2
+0.0,5,7
+0.0,1,1
+0.0,1,1
+0.0,4,5
+0.0,1,1
+0.5,3,4
+0.4330127018922193,3,6
+0.0,1,1
+0.0,1,1
+0.0,21,96
+0.0,11,36
+0.0,1,1
+2.5,8,19
+0.0,1,1
+0.0,1,1
+0.5,3,4
+0.0,1,1
+0.0,2,2
+0.0,5,9
+0.0,1,1
+0.0,5,12
+0.0,1,1
+0.0,3,3
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+4.330127018922194,13,78
+0.0,2,2
+0.0,5,8
+0.0,8,14
+0.4714045207910317,3,5
+0.0,1,1
+0.0,1,1
+1.247219128924647,5,13
+0.0,1,1
+0.0,2,2
+0.0,2,2
+0.0,1,1
+0.0,3,4
+0.0,1,1
+0.816496580927726,4,13
+0.0,9,18
+0.0,1,1
+11.313708498984761,26,148
+0.0,3,3
+0.0,1,1
+0.0,1,1
+4.988876515698588,14,77
+0.0,1,1
+0.0,2,2
+0.0,9,15
+0.0,3,3
+2.0976176963403033,8,30
+0.0,17,115
+0.0,1,1
+0.0,1,1
+0.5,3,4
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,3
+0.0,1,1
+0.5,3,4
+0.5,3,4
+0.0,1,1
+0.816496580927726,4,12
+0.0,1,1
+5.6,16,123
+0.4330127018922193,3,8
+0.0,1,1
+0.0,1,1
+2.165063509461097,8,28
+0.0,1,1
+0.0,1,1
+1.5,5,9
+0.0,17,109
+9.428090415820632,22,210
+2.5,7,16
+0.0,1,1
+0.0,2,2
+0.0,1,1
+4.636809247747852,13,73
+0.0,2,3
+0.0,1,1
+0.0,4,7
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+3.2998316455372216,9,25
+0.0,7,11
+0.0,1,1
+0.0,2,5
+0.0,2,2
+0.8660254037844386,4,9
+0.0,12,40
+0.0,1,1
+0.0,3,3
+0.0,5,8
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+13.5,40,455
+5.5,13,95
+0.0,1,1
+0.0,9,24
+0.0,1,1
+0.0,1,1
+1.224744871391589,5,9
+0.9428090415820634,4,6
+0.0,2,2
+1.4142135623730951,5,7
+0.0,1,1
+0.0,2,2
+2.0,6,16
+0.0,2,2
+1.0,4,6
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.5,3,4
+0.0,7,9
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,2,2
+0.0,5,9
+0.0,1,1
+1.0,4,5
+0.0,1,1
+0.0,2,3
+5.5,13,54
+0.0,1,1
+0.0,1,1
+0.0,4,4
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,3,4
+0.0,1,1
+0.0,7,13
+0.0,1,1
+0.0,1,1
+0.0,1,1
+2.8284271247461903,8,18
+0.0,2,2
+0.0,1,1
+0.9797958971132713,4,10
+0.0,1,1
+0.0,3,3
+0.0,1,1
+0.0,2,3
+0.4330127018922193,3,6
+0.0,1,1
+0.0,1,1
+0.0,1,1
+4.0,17,89
+0.0,2,3
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+2.384848003542364,8,35
+0.0,1,1
+0.0,5,7
+0.0,2,2
+1.0,4,7
+0.0,20,120
+1.7320508075688772,6,15
+0.0,1,1
+0.0,3,3
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,2
+2.5,7,13
+0.0,5,6
+0.0,4,4
+0.0,3,3
+0.0,5,9
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,3,3
+0.0,1,1
+0.0,1,1
+0.0,1,1
+1.5,6,10
+0.0,1,1
+0.0,2,2
+0.0,2,2
+0.0,2,2
+0.0,1,1
+18.958414691194665,69,1909
+0.0,13,29
+0.0,1,1
+0.0,1,1
+0.4714045207910317,3,5
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,2,3
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,4,19
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+1.5,5,10
+0.0,1,1
+0.4714045207910317,3,7
+0.5,3,5
+0.0,1,1
+0.0,2,2
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.5,3,4
+12.5,27,530
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+2.0615528128088303,7,17
+0.0,4,4
+0.0,3,3
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,2,2
+0.0,2,2
+0.0,2,3
+0.0,2,3
+0.0,4,4
+0.0,1,1
+0.0,2,2
+0.0,1,1
+4.5431266766402185,14,114
+0.0,2,2
+0.0,2,3
+1.0,4,9
+0.0,2,2
+0.0,1,1
+0.0,18,126
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,2,2
+0.0,1,1
+3.091206165165235,9,101
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+2.598076211353316,8,21
+3.7712361663282534,10,95
+0.0,43,804
+0.0,1,1
+0.5,3,4
+0.0,2,2
+0.0,3,3
+0.0,2,2
+0.0,2,2
+0.0,1,1
+0.0,3,3
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,16,93
+0.0,1,1
+3.2998316455372216,9,17
+1.5,7,32
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+1.5,5,8
+0.0,1,1
+0.0,1,1
+0.0,7,9
+0.0,4,11
+0.0,7,15
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,1,1
+0.0,2,2
+0.0,1,1
+0.0,3,3
+0.0,1,1
+0.0,1,1
+1.3844373104863459,6,23
+0.0,1,1
+20.98147330914162,48,1551
+0.0,6,9
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1
+0.0,1,1

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: feldtruby
 version: !ruby/object:Gem::Version
-  version: 0.3.10
+  version: 0.3.11
   prerelease: 
 platform: ruby
 authors:
@@ -118,6 +118,7 @@ files:
 - test/test_time.rb
 - test/test_vector.rb
 - test/test_word_counter.rb
+- test/tmp.csv
 homepage: https://github.com/robertfeldt/feldtruby
 licenses: []
 post_install_message: 
@@ -167,3 +168,4 @@ test_files:
 - test/test_time.rb
 - test/test_vector.rb
 - test/test_word_counter.rb
+- test/tmp.csv