rover-df 0.3.1 → 0.3.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e7acc7d0f290ab905594bac7ab1641572e717346f1d0546b3d8b783bcd1e5aca
-  data.tar.gz: 2569b64c237ef836884be2077e8bb8107512d6946631fdb217fb276439412419
+  metadata.gz: c77203874c704a18978e6a2483a1718c65fc96c47ef4dbe178a23bcd8ea6ce32
+  data.tar.gz: 24f04572c8197898387037c6800f487066ad4e56a35ef31df1877b6aef442fcd
 SHA512:
-  metadata.gz: dc46c174fced1c55a96fff28a5c69dfe5a508de527432dac3801cf65375093e290860112191444ceaa23023456452f9331406cdf360a7e689f0db27c289c157c
-  data.tar.gz: 770174c6fb1cc8a52cd47ef77eae871cb5a346d752e9c1c13b8cb44fdc5d791b60fea3d5480bcaa4453b3bca1e83e32af57a57ff0c2a319308d7206d620fa75b
+  metadata.gz: 383e162cd6f62d1d409b869081f4ba97025dd297567a01d300672451a0883f90464f1396b348b6991d1b99a383d4d328f33f8c7cd560e810c96e0ed2f231b688
+  data.tar.gz: 45bb86dfc58172021e7fdef8677f74bbb32075bfe15022e6c0cfe2ddbae0be9b0d1909f5ff42a7792a7863d54bb53afeb309610d22ae0fc02f93f9026c2ab782

data/CHANGELOG.md

@@ -1,3 +1,25 @@
+## 0.3.4 (2022-07-12)
+
+- Added `cbrt`, `erf`, `erfc`, and `hypot` methods to vectors
+- Added `frexp` and `ldexp` methods to vectors
+- Added `base` argument to `log` method
+
+## 0.3.3 (2022-07-11)
+
+- Added `ln`, `log`, `log10`, and `log2` methods to vectors
+- Added `exp` and `exp2` methods to vectors
+- Added `sin`, `cos`, `tan`, `asin`, `acos`, and `atan` methods to vectors
+- Added `sinh`, `cosh`, `tanh`, `asinh`, `acosh`, and `atanh` methods to vectors
+- Added `round`, `ceil`, and `floor` methods to vectors
+- Added `empty?` method to vectors
+- Added `rename` method to data frames
+
+## 0.3.2 (2022-07-10)
+
+- Added `sqrt` method to vectors
+- Improved numeric operations between scalars and vectors
+- Improved performance of `tally`
+
 ## 0.3.1 (2022-05-18)
 
 - Added `to!` to vectors

data/README.md

@@ -191,6 +191,62 @@ df[:a] * 5
 df[:a] / 5
 df[:a] % 5
 df[:a] ** 2
+df[:a].sqrt
+df[:a].cbrt
+df[:a].abs
+```
+
+Rounding
+
+```ruby
+df[:a].round
+df[:a].ceil
+df[:a].floor
+```
+
+Logarithm
+
+```ruby
+df[:a].ln # or log
+df[:a].log(5)
+df[:a].log10
+df[:a].log2
+```
+
+Exponentiation
+
+```ruby
+df[:a].exp
+df[:a].exp2
+```
+
+Trigonometric functions
+
+```ruby
+df[:a].sin
+df[:a].cos
+df[:a].tan
+df[:a].asin
+df[:a].acos
+df[:a].atan
+```
+
+Hyperbolic functions
+
+```ruby
+df[:a].sinh
+df[:a].cosh
+df[:a].tanh
+df[:a].asinh
+df[:a].acosh
+df[:a].atanh
+```
+
+Error function
+
+```ruby
+df[:a].erf
+df[:a].erfc
 ```
 
 Summary statistics
@@ -323,9 +379,11 @@ df.delete(:a)
 df.except!(:a, :b)
 ```
 
-Rename a column
+Rename columns
 
 ```ruby
+df.rename(a: :new_a, b: :new_b)
+# or
 df[:new_a] = df.delete(:a)
 ```
 
@@ -456,7 +514,7 @@ df[:a].type
 Change the type of a column
 
 ```ruby
-df[:a] = df[:a].to(:int32)
+df[:a].to!(:int32)
 ```
 
 ## History

data/lib/rover/data_frame.rb

@@ -136,6 +136,17 @@ module Rover
     alias_method :names, :keys
     alias_method :vector_names, :keys
 
+    def rename(mapping)
+      mapping.each_key do |k|
+        check_column(k)
+      end
+      # use transform_keys! to preserve order
+      @vectors.transform_keys! do |k|
+        mapping[k] || k
+      end
+      self
+    end
+
     def delete(key)
       @vectors.delete(key)
     end

data/lib/rover/vector.rb

@@ -21,6 +21,8 @@ module Rover
       uint: Numo::UInt64
     }
 
+    NOT_SET = Object.new
+
     def initialize(data, type: nil)
       @data = cast_data(data, type: type)
       raise ArgumentError, "Bad size: #{@data.shape}" unless @data.ndim == 1
@@ -87,6 +89,7 @@ module Rover
       if v.is_a?(Vector)
         Vector.new(v.to_numo.mask(@data))
       else
+        # TODO return vector unless v is an integer in 0.4.0
         @data[v]
       end
     end
@@ -184,13 +187,9 @@ module Rover
       Vector.new(@data.to_a.reject(&block))
     end
 
+    # use Ruby tally for performance
     def tally
-      result = Hash.new(0)
-      @data.each do |v|
-        result[v] += 1
-      end
-      result.default = nil
-      result
+      @data.to_a.tally
     end
 
     def sort
@@ -201,6 +200,67 @@ module Rover
       Vector.new(@data.abs)
     end
 
+    def round(ndigits = 0)
+      if ndigits == 0
+        Vector.new(@data.round)
+      else
+        # TODO pass type
+        Vector.new(@data.to_a.map { |v| v.round(ndigits) })
+      end
+    end
+
+    def ceil(ndigits = 0)
+      if ndigits == 0
+        Vector.new(@data.ceil)
+      else
+        # TODO pass type
+        Vector.new(@data.to_a.map { |v| v.ceil(ndigits) })
+      end
+    end
+
+    def floor(ndigits = 0)
+      if ndigits == 0
+        Vector.new(@data.floor)
+      else
+        # TODO pass type
+        Vector.new(@data.to_a.map { |v| v.floor(ndigits) })
+      end
+    end
+
+    [:sqrt, :cbrt, :sin, :cos, :tan, :asin, :acos, :atan, :sinh, :cosh, :tanh, :asinh, :acosh, :atanh, :log2, :log10, :exp, :exp2, :erf, :erfc].each do |m|
+      define_method(m) do
+        Vector.new(Numo::NMath.send(m, @data))
+      end
+    end
+
+    def log(base = NOT_SET)
+      if base == NOT_SET
+        Vector.new(Numo::NMath.log(@data))
+      else
+        type = self.type == :float32 ? :float32 : :float64
+        Vector.new(@data.to_a.map { |v| Math.log(v, base) }, type: type)
+      end
+    end
+
+    def ln
+      log
+    end
+
+    def hypot(y)
+      y = y.to_numo if y.is_a?(Rover::Vector)
+      Vector.new(Numo::NMath.hypot(@data, y))
+    end
+
+    def frexp
+      fraction, exponent = Numo::NMath.frexp(@data)
+      [Vector.new(fraction), Vector.new(exponent)]
+    end
+
+    def ldexp(exponent)
+      exponent = exponent.to_numo if exponent.is_a?(Rover::Vector)
+      Vector.new(Numo::NMath.ldexp(@data, exponent))
+    end
+
     def each(&block)
       @data.each(&block)
     end
@@ -255,10 +315,15 @@ module Rover
       to_a.any?(&block)
     end
 
+    def empty?
+      size == 0
+    end
+
     def zip(other, &block)
       to_a.zip(other.to_a, &block)
     end
 
+    # TODO return element instead of vector if no argument in 0.4.0
     def first(n = 1)
       if n >= size
         Vector.new(@data)
@@ -267,6 +332,7 @@ module Rover
       end
     end
 
+    # TODO return element instead of vector if no argument in 0.4.0
     def last(n = 1)
       if n >= size
         Vector.new(@data)
@@ -350,6 +416,14 @@ module Rover
       super
     end
 
+    def coerce(other)
+      if other.is_a?(Numeric)
+        [Vector.new([other]), self]
+      else
+        raise TypeError, "#{self.class} can't be coerced into #{other.class}"
+      end
+    end
+
     def cast_data(data, type: nil)
       numo_type = numo_type(type) if type
 

data/lib/rover/version.rb

@@ -1,3 +1,3 @@
 module Rover
-  VERSION = "0.3.1"
+  VERSION = "0.3.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rover-df
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.4
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-05-18 00:00:00.000000000 Z
+date: 2022-07-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: numo-narray