polars-df 0.1.4 → 0.1.5

data/lib/polars/series.rb

@@ -23,7 +23,7 @@ module Polars
     # @example Constructing a Series by specifying name and values positionally:
     #   s = Polars::Series.new("a", [1, 2, 3])
     #
-    # @example Notice that the dtype is automatically inferred as a polars Int64:
+    # @example Notice that the dtype is automatically inferred as a polars `:i64`:
     #   s.dtype
     #   # => :i64
     #
@@ -45,6 +45,11 @@ module Polars
 
       name = "" if name.nil?
 
+      # TODO improve
+      if values.is_a?(Range) && values.begin.is_a?(String)
+        values = values.to_a
+      end
+
       if values.nil?
         self._s = sequence_to_rbseries(name, [], dtype: dtype, dtype_if_empty: dtype_if_empty)
       elsif values.is_a?(Series)
@@ -1668,8 +1673,32 @@ module Polars
       super
     end
 
-    # def to_physical
-    # end
+    # Cast to physical representation of the logical dtype.
+    #
+    # - `:date` -> `:i32`
+    # - `:datetime` -> `:i64`
+    # - `:time` -> `:i64`
+    # - `:duration` -> `:i64`
+    # - `:cat` -> `:u32`
+    # - other data types will be left unchanged.
+    #
+    # @return [Series]
+    #
+    # @example
+    #   s = Polars::Series.new("values", ["a", nil, "x", "a"])
+    #   s.cast(:cat).to_physical
+    #   # =>
+    #   # shape: (4,)
+    #   # Series: 'values' [u32]
+    #   # [
+    #   #         0
+    #   #         null
+    #   #         1
+    #   #         0
+    #   # ]
+    def to_physical
+      super
+    end
 
     # Convert this Series to a Ruby Array. This operation clones data.
     #
@@ -1785,8 +1814,34 @@ module Polars
     # def to_numo
     # end
 
-    # def set
-    # end
+    # Set masked values.
+    #
+    # @param filter [Series]
+    #   Boolean mask.
+    # @param value [Object]
+    #   Value with which to replace the masked values.
+    #
+    # @return [Series]
+    #
+    # @note
+    #   Use of this function is frequently an anti-pattern, as it can
+    #   block optimization (predicate pushdown, etc). Consider using
+    #   `Polars.when(predicate).then(value).otherwise(self)` instead.
+    #
+    # @example
+    #   s = Polars::Series.new("a", [1, 2, 3])
+    #   s.set(s == 2, 10)
+    #   # =>
+    #   # shape: (3,)
+    #   # Series: 'a' [i64]
+    #   # [
+    #   #         1
+    #   #         10
+    #   #         3
+    #   # ]
+    def set(filter, value)
+      Utils.wrap_s(_s.send("set_with_mask_#{dtype}", filter._s, value))
+    end
 
     # Set values at the index locations.
     #
@@ -2286,8 +2341,41 @@ module Polars
       super
     end
 
-    # def apply
-    # end
+    # Apply a custom/user-defined function (UDF) over elements in this Series and
+    # return a new Series.
+    #
+    # If the function returns another datatype, the return_dtype arg should be set,
+    # otherwise the method will fail.
+    #
+    # @param return_dtype [Symbol]
+    #   Output datatype. If none is given, the same datatype as this Series will be
+    #   used.
+    # @param skip_nulls [Boolean]
+    #   Nulls will be skipped and not passed to the Ruby function.
+    #   This is faster because Ruby can be skipped and because we call
+    #   more specialized functions.
+    #
+    # @return [Series]
+    #
+    # @example
+    #   s = Polars::Series.new("a", [1, 2, 3])
+    #   s.apply { |x| x + 10 }
+    #   # =>
+    #   # shape: (3,)
+    #   # Series: 'a' [i64]
+    #   # [
+    #   #         11
+    #   #         12
+    #   #         13
+    #   # ]
+    def apply(return_dtype: nil, skip_nulls: true, &func)
+      if return_dtype.nil?
+        pl_return_dtype = nil
+      else
+        pl_return_dtype = Utils.rb_type_to_dtype(return_dtype)
+      end
+      Utils.wrap_s(_s.apply_lambda(func, pl_return_dtype, skip_nulls))
+    end
 
     # Shift the values by a given period.
     #
@@ -2953,8 +3041,35 @@ module Polars
       end
     end
 
-    # def _hash
-    # end
+    # Hash the Series.
+    #
+    # The hash value is of type `:u64`.
+    #
+    # @param seed [Integer]
+    #   Random seed parameter. Defaults to 0.
+    # @param seed_1 [Integer]
+    #   Random seed parameter. Defaults to `seed` if not set.
+    # @param seed_2 [Integer]
+    #   Random seed parameter. Defaults to `seed` if not set.
+    # @param seed_3 [Integer]
+    #   Random seed parameter. Defaults to `seed` if not set.
+    #
+    # @return [Series]
+    #
+    # @example
+    #   s = Polars::Series.new("a", [1, 2, 3])
+    #   s._hash(42)
+    #   # =>
+    #   # shape: (3,)
+    #   # Series: 'a' [u64]
+    #   # [
+    #   #         2374023516666777365
+    #   #         10386026231460783898
+    #   #         17796317186427479491
+    #   # ]
+    def _hash(seed = 0, seed_1 = nil, seed_2 = nil, seed_3 = nil)
+      super
+    end
 
     # Reinterpret the underlying bits as a signed/unsigned integer.
     #
@@ -3424,7 +3539,14 @@ module Polars
         return Utils.wrap_s(_s.send(op, other._s))
       end
 
-      raise Todo
+      if other.is_a?(Date) || other.is_a?(DateTime) || other.is_a?(Time) || other.is_a?(String)
+        raise Todo
+      end
+      if other.is_a?(Float) && !is_float
+        raise Todo
+      end
+
+      Utils.wrap_s(_s.send("#{op}_#{dtype}", other))
     end
 
     def series_to_rbseries(name, values)

data/lib/polars/utils.rb

@@ -11,6 +11,10 @@ module Polars
       DataFrame._from_rbdf(df)
     end
 
+    def self.wrap_ldf(ldf)
+      LazyFrame._from_rbldf(ldf)
+    end
+
     def self.wrap_expr(rbexpr)
       Expr._from_rbexpr(rbexpr)
     end
@@ -171,5 +175,17 @@ module Polars
     def self.bool?(value)
       value == true || value == false
     end
+
+    def self._is_iterable_of(val, eltype)
+      val.all? { |x| x.is_a?(eltype) }
+    end
+
+    def self.is_str_sequence(val, allow_str: false)
+      if allow_str == false && val.is_a?(String)
+        false
+      else
+        val.is_a?(Array) && _is_iterable_of(val, String)
+      end
+    end
   end
 end

data/lib/polars/version.rb

@@ -1,4 +1,4 @@
 module Polars
   # @private
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

data/lib/polars.rb

@@ -1,5 +1,9 @@
 # ext
-require "polars/polars"
+begin
+  require "polars/#{RUBY_VERSION.to_f}/polars"
+rescue LoadError
+  require "polars/polars"
+end
 
 # stdlib
 require "date"
@@ -9,9 +13,11 @@ require "polars/expr_dispatch"
 require "polars/batched_csv_reader"
 require "polars/cat_expr"
 require "polars/cat_name_space"
+require "polars/convert"
 require "polars/data_frame"
 require "polars/date_time_expr"
 require "polars/date_time_name_space"
+require "polars/dynamic_group_by"
 require "polars/exceptions"
 require "polars/expr"
 require "polars/functions"
@@ -23,6 +29,7 @@ require "polars/lazy_group_by"
 require "polars/list_expr"
 require "polars/list_name_space"
 require "polars/meta_expr"
+require "polars/rolling_group_by"
 require "polars/series"
 require "polars/slice"
 require "polars/string_expr"
@@ -35,6 +42,7 @@ require "polars/when"
 require "polars/when_then"
 
 module Polars
+  extend Convert
   extend Functions
   extend IO
   extend LazyFunctions

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: polars-df
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-12-02 00:00:00.000000000 Z
+date: 2022-12-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rb_sys
@@ -39,6 +39,9 @@ files:
 - README.md
 - ext/polars/Cargo.toml
 - ext/polars/extconf.rb
+- ext/polars/src/apply/dataframe.rs
+- ext/polars/src/apply/mod.rs
+- ext/polars/src/apply/series.rs
 - ext/polars/src/batched_csv.rs
 - ext/polars/src/conversion.rs
 - ext/polars/src/dataframe.rs
@@ -60,9 +63,11 @@ files:
 - lib/polars/batched_csv_reader.rb
 - lib/polars/cat_expr.rb
 - lib/polars/cat_name_space.rb
+- lib/polars/convert.rb
 - lib/polars/data_frame.rb
 - lib/polars/date_time_expr.rb
 - lib/polars/date_time_name_space.rb
+- lib/polars/dynamic_group_by.rb
 - lib/polars/exceptions.rb
 - lib/polars/expr.rb
 - lib/polars/expr_dispatch.rb
@@ -75,6 +80,7 @@ files:
 - lib/polars/list_expr.rb
 - lib/polars/list_name_space.rb
 - lib/polars/meta_expr.rb
+- lib/polars/rolling_group_by.rb
 - lib/polars/series.rb
 - lib/polars/slice.rb
 - lib/polars/string_expr.rb
@@ -104,7 +110,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.7
+rubygems_version: 3.3.26
 signing_key:
 specification_version: 4
 summary: Blazingly fast DataFrames for Ruby