polars-df 0.4.0 → 0.5.0

data/lib/polars/struct_name_space.rb

@@ -60,5 +60,37 @@ module Polars
     def rename_fields(names)
       super
     end
+
+    # Get the struct definition as a name/dtype schema dict.
+    #
+    # @return [Object]
+    def schema
+      if _s.nil?
+        {}
+      else
+        _s.dtype.to_schema
+      end
+    end
+
+    # Convert this struct Series to a DataFrame with a separate column for each field.
+    #
+    # @return [DataFrame]
+    #
+    # @example
+    #   s = Polars::Series.new([{"a" => 1, "b" => 2}, {"a" => 3, "b" => 4}])
+    #   s.struct.unnest
+    #   # =>
+    #   # shape: (2, 2)
+    #   # ┌─────┬─────┐
+    #   # │ a   ┆ b   │
+    #   # │ --- ┆ --- │
+    #   # │ i64 ┆ i64 │
+    #   # ╞═════╪═════╡
+    #   # │ 1   ┆ 2   │
+    #   # │ 3   ┆ 4   │
+    #   # └─────┴─────┘
+    def unnest
+      Utils.wrap_df(_s.struct_unnest)
+    end
   end
 end

data/lib/polars/utils.rb

@@ -23,24 +23,42 @@ module Polars
       Polars.col(name)
     end
 
+    def self.arrlen(obj)
+      if obj.is_a?(Range)
+        # size only works for numeric ranges
+        obj.to_a.length
+      elsif obj.is_a?(String)
+        nil
+      else
+        obj.length
+      end
+    rescue
+      nil
+    end
+
     def self._timedelta_to_pl_duration(td)
       td
     end
 
     def self._datetime_to_pl_timestamp(dt, tu)
       if tu == "ns"
-        (dt.to_datetime.utc.to_f * 1e9).to_i
+        (dt.to_datetime.to_time.to_f * 1e9).to_i
       elsif tu == "us"
-        (dt.to_datetime.utc.to_f * 1e6).to_i
+        (dt.to_datetime.to_time.to_f * 1e6).to_i
       elsif tu == "ms"
-        (dt.to_datetime.utc.to_f * 1e3).to_i
+        (dt.to_datetime.to_time.to_f * 1e3).to_i
       elsif tu.nil?
-        (dt.to_datetime.utc.to_f * 1e6).to_i
+        (dt.to_datetime.to_time.to_f * 1e6).to_i
       else
         raise ArgumentError, "tu must be one of {{'ns', 'us', 'ms'}}, got #{tu}"
       end
     end
 
+    def self._date_to_pl_date(d)
+      dt = d.to_datetime.to_time
+      dt.to_i / (3600 * 24)
+    end
+
     def self._to_ruby_datetime(value, dtype, tu: "ns", tz: nil)
       if dtype == :date || dtype == Date
         # days to seconds
@@ -69,6 +87,18 @@ module Polars
       end
     end
 
+    def self._to_ruby_duration(value, tu = "ns")
+      if tu == "ns"
+        value / 1e9
+      elsif tu == "us"
+        value / 1e6
+      elsif tu == "ms"
+        value / 1e3
+      else
+        raise ArgumentError, "tu must be one of {{'ns', 'us', 'ms'}}, got #{tu}"
+      end
+    end
+
     def self.selection_to_rbexpr_list(exprs)
       if exprs.is_a?(String) || exprs.is_a?(Symbol) || exprs.is_a?(Expr) || exprs.is_a?(Series)
         exprs = [exprs]
@@ -102,7 +132,10 @@ module Polars
     end
 
     # TODO fix
-    def self.is_polars_dtype(data_type)
+    def self.is_polars_dtype(data_type, include_unknown: false)
+      if data_type == Unknown
+        return include_unknown
+      end
       data_type.is_a?(Symbol) || data_type.is_a?(String) || data_type.is_a?(DataType) || (data_type.is_a?(Class) && data_type < DataType)
     end
 
@@ -113,7 +146,8 @@ module Polars
       TrueClass => :bool,
       FalseClass => :bool,
       ::Date => :date,
-      ::DateTime => :datetime
+      ::DateTime => :datetime,
+      ::Time => :datetime
     }
 
     # TODO fix
@@ -178,7 +212,7 @@ module Polars
     end
 
     def self.bool?(value)
-      value == true || value == false
+      value.is_a?(TrueClass) || value.is_a?(FalseClass)
     end
 
     def self.strlike?(value)

data/lib/polars/version.rb

@@ -1,4 +1,4 @@
 module Polars
   # @private
-  VERSION = "0.4.0"
+  VERSION = "0.5.0"
 end

data/lib/polars.rb

@@ -1,8 +1,8 @@
 # ext
 begin
-  require_relative "polars/#{RUBY_VERSION.to_f}/polars"
+  require "polars/#{RUBY_VERSION.to_f}/polars"
 rescue LoadError
-  require_relative "polars/polars"
+  require "polars/polars"
 end
 
 # stdlib

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: polars-df
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-04-01 00:00:00.000000000 Z
+date: 2023-05-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rb_sys
@@ -40,27 +40,42 @@ files:
 - ext/polars/Cargo.toml
 - ext/polars/extconf.rb
 - ext/polars/src/apply/dataframe.rs
+- ext/polars/src/apply/lazy.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
 - ext/polars/src/error.rs
+- ext/polars/src/expr.rs
+- ext/polars/src/expr/binary.rs
+- ext/polars/src/expr/categorical.rs
+- ext/polars/src/expr/datetime.rs
+- ext/polars/src/expr/general.rs
+- ext/polars/src/expr/list.rs
+- ext/polars/src/expr/meta.rs
+- ext/polars/src/expr/string.rs
+- ext/polars/src/expr/struct.rs
 - ext/polars/src/file.rs
-- ext/polars/src/lazy/apply.rs
-- ext/polars/src/lazy/dataframe.rs
-- ext/polars/src/lazy/dsl.rs
-- ext/polars/src/lazy/meta.rs
-- ext/polars/src/lazy/mod.rs
-- ext/polars/src/lazy/utils.rs
+- ext/polars/src/functions/eager.rs
+- ext/polars/src/functions/io.rs
+- ext/polars/src/functions/lazy.rs
+- ext/polars/src/functions/meta.rs
+- ext/polars/src/functions/mod.rs
+- ext/polars/src/functions/whenthen.rs
+- ext/polars/src/lazyframe.rs
+- ext/polars/src/lazygroupby.rs
 - ext/polars/src/lib.rs
-- ext/polars/src/list_construction.rs
-- ext/polars/src/numo.rs
 - ext/polars/src/object.rs
 - ext/polars/src/prelude.rs
 - ext/polars/src/rb_modules.rs
 - ext/polars/src/series.rs
-- ext/polars/src/set.rs
+- ext/polars/src/series/aggregation.rs
+- ext/polars/src/series/arithmetic.rs
+- ext/polars/src/series/comparison.rs
+- ext/polars/src/series/construction.rs
+- ext/polars/src/series/export.rs
+- ext/polars/src/series/set_at_idx.rs
 - ext/polars/src/utils.rs
 - lib/polars-df.rb
 - lib/polars.rb