rover-df 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1b269feff9f961b17f921d168f240e24cea8e02a75438cdfc3e0aff5398d78f4
-  data.tar.gz: dcd83d0a8ebe2ed3b13c1fdfca8c1a078d3a2b12df66a1c59ebfecf7030ee443
+  metadata.gz: fda55b893da90eb87f53722b328d5150649ce9d91d1d49be1a9523a11229ab30
+  data.tar.gz: 1b18963c432f5489a1f89cf88d0cd509d4ec6a28d4b0bb2e5c400d66f7d2921e
 SHA512:
-  metadata.gz: 74743dad81d9fd28a96818e909f165ec173241d04ef9daeb49b8fbc48ff02055d1946dddf33d052fe68f59c32ac678225c5eef9f1d03a035c35630fa798fda26
-  data.tar.gz: d1b12368ae872abfff95cfffaf209e9ccb05b121aaf1f9b23b11785cc00fa9c3915ec15905275c1c69cbcbaefe1f1cae494af648d3ef3a2c9af58b6b931152d5
+  metadata.gz: 0d69590856829d59616750b3912579f9cf78b1ae33b9a9448cf72fa26f45596225847f6800b3c4a2bddb8d9390a280618de98457c76cc7be0f236f8844f4d4bd
+  data.tar.gz: af77b30aac473131c11994fafa9784580a33bf69da2422baad580a193bb3ea7a627b40430ab8295bb6b6330d30bfc11fda8bda6874379b9573794dc1ed38b4c9

data/CHANGELOG.md

@@ -1,3 +1,11 @@
+## 1.1.0 (2026-08-15)
+
+- Added support for integer vectors to `[]` method
+
+## 1.0.1 (2026-04-15)
+
+- Added support for positional argument to `any?` and `all?` methods
+
 ## 1.0.0 (2026-04-04)
 
 - Switched to `numo-narray-alt`

data/README.md

@@ -2,7 +2,7 @@
 
 Simple, powerful data frames for Ruby
 
-:mountain: Designed for data exploration and machine learning, and powered by [Numo](https://github.com/ruby-numo/numo-narray)
+:mountain: Designed for data exploration and machine learning, and powered by [numo-narray-alt](https://github.com/yoshoku/numo-narray-alt)
 
 :evergreen_tree: Uses [Vega](https://github.com/ankane/vega) for visualization
 

data/lib/rover/data_frame.rb

@@ -27,7 +27,7 @@ module Rover
         end
       elsif data.is_a?(Array)
         vectors = {}
-        raise ArgumentError, "Array elements must be hashes" unless data.all? { |d| d.is_a?(Hash) }
+        raise ArgumentError, "Array elements must be hashes" unless data.all?(Hash)
         keys = data.flat_map(&:keys).uniq
         keys.each do |k|
           vectors[k] = []
@@ -65,13 +65,19 @@ module Rover
     end
 
     def [](where)
-      if (where.is_a?(Vector) && where.to_numo.is_a?(Numo::Bit)) || where.is_a?(Numeric) || where.is_a?(Range) || (where.is_a?(Array) && where.all? { |v| v.is_a?(Integer) })
+      # TODO use path for all vectors in 2.0
+      if (where.is_a?(Vector) && (where.type == :bool || Vector::INTEGER_TYPES.include?(where.type))) || where.is_a?(Numeric) || where.is_a?(Range) || (where.is_a?(Array) && where.all?(Integer))
         new_vectors = {}
         @vectors.each do |k, v|
           new_vectors[k] = v[where]
         end
         DataFrame.new(new_vectors)
       elsif where.is_a?(Array)
+        # TODO raise error in 2.0
+        unless where.all? { |v| v.is_a?(String) || v.is_a?(Symbol) }
+          warn "[rover] Unsupported selector"
+        end
+
         # multiple columns
         df = DataFrame.new
         where.each do |k|
@@ -80,6 +86,11 @@ module Rover
         end
         df
       else
+        # TODO raise error in 2.0
+        unless where.is_a?(String) || where.is_a?(Symbol)
+          warn "[rover] Unsupported selector"
+        end
+
         # single column
         @vectors[where.to_s]
       end
@@ -269,7 +280,7 @@ module Rover
           when :bool
             :boolean
           when :object
-            if @vectors[name].all? { |v| v.is_a?(String) }
+            if @vectors[name].all?(String)
               :string
             else
               raise "Unknown type"
@@ -313,10 +324,10 @@ module Rover
 
       @vectors.each do |k, v|
         v = summarize ? v.first(5).to_a + ["..."] + v.last(5).to_a : v.to_a
-        width = ([k] + v).map(&:to_s).map(&:size).max
+        width = ([k] + v).map { |v2| v2.to_s.size }.max
         width = 3 if width < 3
 
-        if lines.empty? || lines[-2].map { |l| l.size + spaces }.sum + width > 120
+        if lines.empty? || lines[-2].sum { |l| l.size + spaces } + width > 120
           line_start = lines.size
           lines << []
           v.size.times do |i|
@@ -382,10 +393,10 @@ module Rover
 
       size = self.size
       vectors.each do |k, v|
-        @vectors[k] = Vector.new(v.to_a + (other[k] ? other[k].to_a : [nil] * other.size))
+        @vectors[k] = Vector.new(v.to_a + (other[k] ? other[k].to_a : Array.new(other.size)))
       end
       (other.vector_names - vector_names).each do |k|
-        @vectors[k] = Vector.new([nil] * size + other[k].to_a)
+        @vectors[k] = Vector.new(Array.new(size) + other[k].to_a)
       end
       self
     end

data/lib/rover/vector.rb

@@ -23,6 +23,9 @@ module Rover
 
     NOT_SET = Object.new
 
+    # private
+    INTEGER_TYPES = [:int8, :int16, :int32, :int64, :uint8, :uint16, :uint32, :uint64]
+
     def initialize(data, type: nil)
       @data = cast_data(data, type: type)
       raise ArgumentError, "Bad size: #{@data.shape}" unless @data.ndim == 1
@@ -87,7 +90,14 @@ module Rover
 
     def [](v)
       if v.is_a?(Vector)
-        Vector.new(v.to_numo.mask(@data))
+        case v.type
+        when :bool
+          Vector.new(v.to_numo.mask(@data))
+        when *INTEGER_TYPES
+          Vector.new(@data[v.to_numo])
+        else
+          raise ArgumentError, "Unsupported selector"
+        end
       elsif v.is_a?(Numeric)
         @data[v]
       else
@@ -305,12 +315,12 @@ module Rover
       @data.cast_to(Numo::DFloat).var
     end
 
-    def all?(&block)
-      to_a.all?(&block)
+    def all?(...)
+      to_a.all?(...)
     end
 
-    def any?(&block)
-      to_a.any?(&block)
+    def any?(...)
+      to_a.any?(...)
     end
 
     def empty?
@@ -374,7 +384,7 @@ module Rover
     end
 
     def one_hot(drop: false, prefix: nil)
-      raise ArgumentError, "All elements must be strings" unless all? { |vi| vi.is_a?(String) }
+      raise ArgumentError, "All elements must be strings" unless all?(String)
 
       new_vectors = {}
       # maybe sort values first
@@ -459,7 +469,7 @@ module Rover
           data = numo_type.cast(data)
         else
           data =
-            if data.all? { |v| v.is_a?(Integer) }
+            if data.all?(Integer)
               Numo::Int64.cast(data)
             elsif data.all? { |v| v.is_a?(Numeric) || v.nil? }
               Numo::DFloat.cast(data.map { |v| v || Float::NAN })

data/lib/rover/version.rb

@@ -1,3 +1,3 @@
 module Rover
-  VERSION = "1.0.0"
+  VERSION = "1.1.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rover-df
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Andrew Kane
@@ -55,7 +55,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.6
+rubygems_version: 4.0.14
 specification_version: 4
 summary: Simple, powerful data frames for Ruby
 test_files: []