davidrichards-data_frame 0.0.8 → 0.0.9

data/VERSION.yml

@@ -1,4 +1,4 @@
 --- 
 :major: 0
 :minor: 0
-:patch: 8
+:patch: 9

data/lib/data_frame.rb

@@ -4,6 +4,14 @@ require 'just_enumerable_stats'
 require 'open-uri'
 require 'fastercsv'
 
+# Use a Dictionary if available
+begin
+  require 'facets/dictionary'
+rescue LoadError => e
+  # Do nothing
+end
+
+
 Dir.glob("#{File.dirname(__FILE__)}/ext/*.rb").each { |file| require file }
 
 $:.unshift(File.dirname(__FILE__))
@@ -94,6 +102,25 @@ class DataFrame
     @items.transpose[i]
   end
   
+  # The rows as an array of arrays, an alias for items.
+  alias :rows :items
+  
+  # The columns as a Dictionary or Hash
+  # This is cached, call columns(true) to reset the cache.
+  def columns(reset=false)
+    @columns = nil if reset
+    return @columns if @columns
+    
+    container = defined?(Dictionary) ? Dictionary.new : Hash.new
+    i = 0
+    
+    @columns = @items.transpose.inject(container) do |cont, col|
+      cont[@labels[i]] = col
+      i += 1
+      cont
+    end
+  end
+  
   def render_row(sym)
     i = self.row_labels.index(sym)
     return nil unless i

data/spec/data_frame_spec.rb

@@ -101,5 +101,18 @@ describe DataFrame do
     @df.labels.should eql([:threes, :fours])
   end
   
+  it "should offer a hash-like structure of columns" do
+    @df.add [1,2,3,4]
+    @df.add [5, 6, 7, 8]
+    @df.columns[:these].should eql([1, 5])
+    @df.columns[:are].should eql([2, 6])
+    @df.columns[:the].should eql([3, 7])
+    @df.columns[:labels].should eql([4, 8])
+  end
   
+  it "should alias items with rows" do
+    @df.add [1,2,3,4]
+    @df.add [5, 6, 7, 8]
+    @df.rows.should eql(@df.items)
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: davidrichards-data_frame
 version: !ruby/object:Gem::Version 
-  version: 0.0.8
+  version: 0.0.9
 platform: ruby
 authors: 
 - David Richards
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-08-03 00:00:00 -07:00
+date: 2009-08-04 00:00:00 -07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency