polars-df 0.1.3 → 0.1.5

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

@@ -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
@@ -37,6 +41,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]
@@ -143,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.3"
+  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"
@@ -8,8 +12,12 @@ require "date"
 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"
@@ -19,17 +27,22 @@ 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/rolling_group_by"
 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"
 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.3
+  version: 0.1.5
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-11-28 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
@@ -51,6 +54,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 +62,12 @@ files:
 - lib/polars.rb
 - 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
@@ -70,11 +78,15 @@ 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/rolling_group_by.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
@@ -98,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