polars-df 0.1.3 → 0.1.4

data/lib/polars/struct_name_space.rb

@@ -0,0 +1,64 @@
+module Polars
+  # Series.struct namespace.
+  class StructNameSpace
+    include ExprDispatch
+
+    self._accessor = "struct"
+
+    # @private
+    def initialize(series)
+      self._s = series._s
+    end
+
+    # Retrieve one of the fields of this `Struct` as a new Series.
+    #
+    # @return [Series]
+    def [](item)
+      if item.is_a?(Integer)
+        field(fields[item])
+      elsif item.is_a?(String)
+        field(item)
+      else
+        raise ArgumentError, "expected type Integer or String, got #{item.class.name}"
+      end
+    end
+
+    # Convert this Struct Series to a DataFrame.
+    #
+    # @return [DataFrame]
+    def to_frame
+      Utils.wrap_df(_s.struct_to_frame)
+    end
+
+    # Get the names of the fields.
+    #
+    # @return [Array]
+    def fields
+      if _s.nil?
+        []
+      else
+        _s.struct_fields
+      end
+    end
+
+    # Retrieve one of the fields of this `Struct` as a new Series.
+    #
+    # @param name [String]
+    #   Name of the field
+    #
+    # @return [Series]
+    def field(name)
+      super
+    end
+
+    # Rename the fields of the struct.
+    #
+    # @param names [Array]
+    #   New names in the order of the struct's fields
+    #
+    # @return [Series]
+    def rename_fields(names)
+      super
+    end
+  end
+end

data/lib/polars/utils.rb

@@ -37,6 +37,34 @@ module Polars
       end
     end
 
+    def self._to_ruby_datetime(value, dtype, tu: "ns", tz: nil)
+      if dtype == :date
+        # days to seconds
+        # important to create from utc. Not doing this leads
+        # to inconsistencies dependent on the timezone you are in.
+        Time.at(value * 86400).utc.to_date
+      # TODO fix dtype
+      elsif dtype.to_s.start_with?("datetime[")
+        if tz.nil? || tz == ""
+          if tu == "ns"
+            raise Todo
+          elsif tu == "us"
+            dt = Time.at(value / 1000000, value % 1000000, :usec).utc
+          elsif tu == "ms"
+            raise Todo
+          else
+            raise ArgumentError, "tu must be one of {{'ns', 'us', 'ms'}}, got #{tu}"
+          end
+        else
+          raise Todo
+        end
+
+        dt
+      else
+        raise NotImplementedError
+      end
+    end
+
     def self.selection_to_rbexpr_list(exprs)
       if exprs.is_a?(String) || exprs.is_a?(Expr) || exprs.is_a?(Series)
         exprs = [exprs]

data/lib/polars/version.rb

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

data/lib/polars.rb

@@ -8,8 +8,10 @@ require "date"
 require "polars/expr_dispatch"
 require "polars/batched_csv_reader"
 require "polars/cat_expr"
+require "polars/cat_name_space"
 require "polars/data_frame"
 require "polars/date_time_expr"
+require "polars/date_time_name_space"
 require "polars/exceptions"
 require "polars/expr"
 require "polars/functions"
@@ -19,11 +21,14 @@ require "polars/lazy_frame"
 require "polars/lazy_functions"
 require "polars/lazy_group_by"
 require "polars/list_expr"
+require "polars/list_name_space"
 require "polars/meta_expr"
 require "polars/series"
 require "polars/slice"
 require "polars/string_expr"
+require "polars/string_name_space"
 require "polars/struct_expr"
+require "polars/struct_name_space"
 require "polars/utils"
 require "polars/version"
 require "polars/when"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: polars-df
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-11-28 00:00:00.000000000 Z
+date: 2022-12-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rb_sys
@@ -51,6 +51,7 @@ files:
 - ext/polars/src/lazy/mod.rs
 - ext/polars/src/lazy/utils.rs
 - ext/polars/src/lib.rs
+- ext/polars/src/list_construction.rs
 - ext/polars/src/series.rs
 - ext/polars/src/set.rs
 - ext/polars/src/utils.rs
@@ -58,8 +59,10 @@ files:
 - lib/polars.rb
 - lib/polars/batched_csv_reader.rb
 - lib/polars/cat_expr.rb
+- lib/polars/cat_name_space.rb
 - lib/polars/data_frame.rb
 - lib/polars/date_time_expr.rb
+- lib/polars/date_time_name_space.rb
 - lib/polars/exceptions.rb
 - lib/polars/expr.rb
 - lib/polars/expr_dispatch.rb
@@ -70,11 +73,14 @@ files:
 - lib/polars/lazy_functions.rb
 - lib/polars/lazy_group_by.rb
 - lib/polars/list_expr.rb
+- lib/polars/list_name_space.rb
 - lib/polars/meta_expr.rb
 - lib/polars/series.rb
 - lib/polars/slice.rb
 - lib/polars/string_expr.rb
+- lib/polars/string_name_space.rb
 - lib/polars/struct_expr.rb
+- lib/polars/struct_name_space.rb
 - lib/polars/utils.rb
 - lib/polars/version.rb
 - lib/polars/when.rb