polars-df 0.9.0-x86_64-linux → 0.11.0-x86_64-linux

data/lib/polars/string_expr.rb

@@ -840,6 +840,7 @@ module Polars
     #   # │ true     │
     #   # └──────────┘
     def json_path_match(json_path)
+      json_path = Utils.parse_as_expression(json_path, str_as_lit: true)
       Utils.wrap_expr(_rbexpr.str_json_path_match(json_path))
     end
 
@@ -1018,15 +1019,15 @@ module Polars
     #   )
     #   # =>
     #   # shape: (3, 3)
-    #   # ┌───────────────────────────────────┬───────────────────────┬──────────┐
-    #   # │ url                               ┆ captures              ┆ name     │
-    #   # │ ---                               ┆ ---                   ┆ ---      │
-    #   # │ str                               ┆ struct[2]             ┆ str      │
-    #   # ╞═══════════════════════════════════╪═══════════════════════╪══════════╡
-    #   # │ http://vote.com/ballon_dor?candi… ┆ {"messi","python"}    ┆ MESSI    │
-    #   # │ http://vote.com/ballon_dor?candi… ┆ {"weghorst","polars"} ┆ WEGHORST │
-    #   # │ http://vote.com/ballon_dor?error… ┆ {null,null}           ┆ null     │
-    #   # └───────────────────────────────────┴───────────────────────┴──────────┘
+    #   # ┌─────────────────────────────────┬───────────────────────┬──────────┐
+    #   # │ url                             ┆ captures              ┆ name     │
+    #   # │ ---                             ┆ ---                   ┆ ---      │
+    #   # │ str                             ┆ struct[2]             ┆ str      │
+    #   # ╞═════════════════════════════════╪═══════════════════════╪══════════╡
+    #   # │ http://vote.com/ballon_dor?can… ┆ {"messi","python"}    ┆ MESSI    │
+    #   # │ http://vote.com/ballon_dor?can… ┆ {"weghorst","polars"} ┆ WEGHORST │
+    #   # │ http://vote.com/ballon_dor?err… ┆ {null,null}           ┆ null     │
+    #   # └─────────────────────────────────┴───────────────────────┴──────────┘
     def extract_groups(pattern)
       Utils.wrap_expr(_rbexpr.str_extract_groups(pattern))
     end
@@ -1354,6 +1355,7 @@ module Polars
     #   # │ null ┆ null   │
     #   # └──────┴────────┘
     def to_integer(base: 10, strict: true)
+      base = Utils.parse_as_expression(base, str_as_lit: false)
       Utils.wrap_expr(_rbexpr.str_to_integer(base, strict))
     end
 
@@ -1417,15 +1419,15 @@ module Polars
     #   )
     #   # =>
     #   # shape: (3, 2)
-    #   # ┌───────────────────────────────────┬──────────────┐
-    #   # │ lyrics                            ┆ contains_any │
-    #   # │ ---                               ┆ ---          │
-    #   # │ str                               ┆ bool         │
-    #   # ╞═══════════════════════════════════╪══════════════╡
-    #   # │ Everybody wants to rule the worl… ┆ false        │
-    #   # │ Tell me what you want, what you … ┆ true         │
-    #   # │ Can you feel the love tonight     ┆ true         │
-    #   # └───────────────────────────────────┴──────────────┘
+    #   # ┌─────────────────────────────────┬──────────────┐
+    #   # │ lyrics                          ┆ contains_any │
+    #   # │ ---                             ┆ ---          │
+    #   # │ str                             ┆ bool         │
+    #   # ╞═════════════════════════════════╪══════════════╡
+    #   # │ Everybody wants to rule the wo… ┆ false        │
+    #   # │ Tell me what you want, what yo… ┆ true         │
+    #   # │ Can you feel the love tonight   ┆ true         │
+    #   # └─────────────────────────────────┴──────────────┘
     def contains_any(patterns, ascii_case_insensitive: false)
       patterns = Utils.parse_as_expression(patterns, str_as_lit: false, list_as_lit: false)
       Utils.wrap_expr(
@@ -1467,15 +1469,15 @@ module Polars
     #   )
     #   # =>
     #   # shape: (3, 2)
-    #   # ┌───────────────────────────────────┬───────────────────────────────────┐
-    #   # │ lyrics                            ┆ removes_pronouns                  │
-    #   # │ ---                               ┆ ---                               │
-    #   # │ str                               ┆ str                               │
-    #   # ╞═══════════════════════════════════╪═══════════════════════════════════╡
-    #   # │ Everybody wants to rule the worl… ┆ Everybody wants to rule the worl… │
-    #   # │ Tell me what you want, what you … ┆ Tell  what  want, what  really r… │
-    #   # │ Can you feel the love tonight     ┆ Can  feel the love tonight        │
-    #   # └───────────────────────────────────┴───────────────────────────────────┘
+    #   # ┌─────────────────────────────────┬─────────────────────────────────┐
+    #   # │ lyrics                          ┆ removes_pronouns                │
+    #   # │ ---                             ┆ ---                             │
+    #   # │ str                             ┆ str                             │
+    #   # ╞═════════════════════════════════╪═════════════════════════════════╡
+    #   # │ Everybody wants to rule the wo… ┆ Everybody wants to rule the wo… │
+    #   # │ Tell me what you want, what yo… ┆ Tell  what  want, what  really… │
+    #   # │ Can you feel the love tonight   ┆ Can  feel the love tonight      │
+    #   # └─────────────────────────────────┴─────────────────────────────────┘
     #
     # @example
     #   df.with_columns(
@@ -1488,15 +1490,15 @@ module Polars
     #   )
     #   # =>
     #   # shape: (3, 2)
-    #   # ┌───────────────────────────────────┬───────────────────────────────────┐
-    #   # │ lyrics                            ┆ confusing                         │
-    #   # │ ---                               ┆ ---                               │
-    #   # │ str                               ┆ str                               │
-    #   # ╞═══════════════════════════════════╪═══════════════════════════════════╡
-    #   # │ Everybody wants to rule the worl… ┆ Everybody wants to rule the worl… │
-    #   # │ Tell me what you want, what you … ┆ Tell you what me want, what me r… │
-    #   # │ Can you feel the love tonight     ┆ Can me feel the love tonight      │
-    #   # └───────────────────────────────────┴───────────────────────────────────┘
+    #   # ┌─────────────────────────────────┬─────────────────────────────────┐
+    #   # │ lyrics                          ┆ confusing                       │
+    #   # │ ---                             ┆ ---                             │
+    #   # │ str                             ┆ str                             │
+    #   # ╞═════════════════════════════════╪═════════════════════════════════╡
+    #   # │ Everybody wants to rule the wo… ┆ Everybody wants to rule the wo… │
+    #   # │ Tell me what you want, what yo… ┆ Tell you what me want, what me… │
+    #   # │ Can you feel the love tonight   ┆ Can me feel the love tonight    │
+    #   # └─────────────────────────────────┴─────────────────────────────────┘
     def replace_many(patterns, replace_with, ascii_case_insensitive: false)
       patterns = Utils.parse_as_expression(patterns, str_as_lit: false, list_as_lit: false)
       replace_with = Utils.parse_as_expression(

data/lib/polars/utils.rb

@@ -139,7 +139,7 @@ module Polars
       Polars.lit(value)
     end
 
-    def self.normalise_filepath(path, check_not_directory: true)
+    def self.normalize_filepath(path, check_not_directory: true)
       path = File.expand_path(path)
       if check_not_directory && File.exist?(path) && Dir.exist?(path)
         raise ArgumentError, "Expected a file path; #{path} is a directory"
@@ -364,5 +364,93 @@ module Polars
         end
       end
     end
+
+    def self._expand_selectors(frame, *items)
+      items_iter = _parse_inputs_as_iterable(items)
+
+      expanded = []
+      items_iter.each do |item|
+        if is_selector(item)
+          selector_cols = expand_selector(frame, item)
+          expanded.concat(selector_cols)
+        else
+          expanded << item
+        end
+      end
+      expanded
+    end
+
+    # TODO
+    def self.is_selector(obj)
+      false
+    end
+
+    def self.parse_predicates_constraints_as_expression(*predicates, **constraints)
+      all_predicates = _parse_positional_inputs(predicates)
+
+      if constraints.any?
+        constraint_predicates = _parse_constraints(constraints)
+        all_predicates.concat(constraint_predicates)
+      end
+
+      _combine_predicates(all_predicates)
+    end
+
+    def self._parse_constraints(constraints)
+      constraints.map do |name, value|
+        Polars.col(name).eq(value)._rbexpr
+      end
+    end
+
+    def self._combine_predicates(predicates)
+      if !predicates.any?
+        msg = "at least one predicate or constraint must be provided"
+        raise TypeError, msg
+      end
+
+      if predicates.length == 1
+        return predicates[0]
+      end
+
+      Plr.all_horizontal(predicates)
+    end
+
+    def self.parse_when_inputs(*predicates, **constraints)
+      parse_predicates_constraints_as_expression(*predicates, **constraints)
+    end
+
+    def self.parse_interval_argument(interval)
+      if interval.include?(" ")
+        interval = interval.gsub(" ", "")
+      end
+      interval.downcase
+    end
+
+    def self.validate_rolling_by_aggs_arguments(weights, center:)
+      if !weights.nil?
+        msg = "`weights` is not supported in `rolling_*(..., by=...)` expression"
+        raise InvalidOperationError, msg
+      end
+      if center
+        msg = "`center=True` is not supported in `rolling_*(..., by=...)` expression"
+        raise InvalidOperationError, msg
+      end
+    end
+
+    def self.validate_rolling_aggs_arguments(window_size, closed)
+      if window_size.is_a?(::String)
+        begin
+          window_size = window_size.delete_suffix("i").to_i
+        rescue
+          msg = "Expected a string of the form 'ni', where `n` is a positive integer, got: #{window_size}"
+          raise InvalidOperationError, msg
+        end
+      end
+      if !closed.nil?
+        msg = "`closed` is not supported in `rolling_*(...)` expression"
+        raise InvalidOperationError, msg
+      end
+      window_size
+    end
   end
 end

data/lib/polars/version.rb

@@ -1,4 +1,4 @@
 module Polars
   # @private
-  VERSION = "0.9.0"
+  VERSION = "0.11.0"
 end

data/lib/polars/whenthen.rb

@@ -0,0 +1,83 @@
+module Polars
+  # @private
+  class When
+    attr_accessor :_when
+
+    def initialize(rbwhen)
+      self._when = rbwhen
+    end
+
+    def then(statement)
+      statement_rbexpr = Utils.parse_as_expression(statement)
+      Then.new(_when.then(statement_rbexpr))
+    end
+  end
+
+  # @private
+  class Then < Expr
+    attr_accessor :_then
+
+    def initialize(rbthen)
+      self._then = rbthen
+    end
+
+    def self._from_rbexpr(rbexpr)
+      Utils.wrap_expr(rbexpr)
+    end
+
+    def _rbexpr
+      _then.otherwise(Polars.lit(nil)._rbexpr)
+    end
+
+    def when(*predicates, **constraints)
+      condition_rbexpr = Utils.parse_when_inputs(*predicates, **constraints)
+      ChainedWhen.new(_then.when(condition_rbexpr))
+    end
+
+    def otherwise(statement)
+      statement_rbexpr = Utils.parse_as_expression(statement)
+      Utils.wrap_expr(_then.otherwise(statement_rbexpr))
+    end
+  end
+
+  # @private
+  class ChainedWhen
+    attr_accessor :_chained_when
+
+    def initialize(chained_when)
+      self._chained_when = chained_when
+    end
+
+    def then(statement)
+      statement_rbexpr = Utils.parse_as_expression(statement)
+      ChainedThen.new(_chained_when.then(statement_rbexpr))
+    end
+  end
+
+  # @private
+  class ChainedThen < Expr
+    attr_accessor :_chained_then
+
+    def initialize(chained_then)
+      self._chained_then = chained_then
+    end
+
+    def self._from_rbexpr(rbexpr)
+      Utils.wrap_expr(rbexpr)
+    end
+
+    def _rbexpr
+      _chained_then.otherwise(Polars.lit(nil)._rbexpr)
+    end
+
+    def when(*predicates, **constraints)
+      condition_rbexpr = Utils.parse_when_inputs(*predicates, **constraints)
+      ChainedWhen.new(_chained_then.when(condition_rbexpr))
+    end
+
+    def otherwise(statement)
+      statement_rbexpr = Utils.parse_as_expression(statement)
+      Utils.wrap_expr(_chained_then.otherwise(statement_rbexpr))
+    end
+  end
+end

data/lib/polars.rb

@@ -42,9 +42,17 @@ require_relative "polars/functions/whenthen"
 require_relative "polars/functions/aggregation/horizontal"
 require_relative "polars/functions/aggregation/vertical"
 require_relative "polars/functions/range/date_range"
+require_relative "polars/functions/range/datetime_range"
 require_relative "polars/functions/range/int_range"
+require_relative "polars/functions/range/time_range"
 require_relative "polars/group_by"
-require_relative "polars/io"
+require_relative "polars/io/avro"
+require_relative "polars/io/csv"
+require_relative "polars/io/database"
+require_relative "polars/io/ipc"
+require_relative "polars/io/json"
+require_relative "polars/io/ndjson"
+require_relative "polars/io/parquet"
 require_relative "polars/lazy_frame"
 require_relative "polars/lazy_group_by"
 require_relative "polars/list_expr"
@@ -63,8 +71,7 @@ require_relative "polars/struct_name_space"
 require_relative "polars/testing"
 require_relative "polars/utils"
 require_relative "polars/version"
-require_relative "polars/when"
-require_relative "polars/when_then"
+require_relative "polars/whenthen"
 
 module Polars
   extend Convert

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: polars-df
 version: !ruby/object:Gem::Version
-  version: 0.9.0
+  version: 0.11.0
 platform: x86_64-linux
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-03-03 00:00:00.000000000 Z
+date: 2024-06-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bigdecimal
@@ -71,11 +71,19 @@ files:
 - lib/polars/functions/lit.rb
 - lib/polars/functions/random.rb
 - lib/polars/functions/range/date_range.rb
+- lib/polars/functions/range/datetime_range.rb
 - lib/polars/functions/range/int_range.rb
+- lib/polars/functions/range/time_range.rb
 - lib/polars/functions/repeat.rb
 - lib/polars/functions/whenthen.rb
 - lib/polars/group_by.rb
-- lib/polars/io.rb
+- lib/polars/io/avro.rb
+- lib/polars/io/csv.rb
+- lib/polars/io/database.rb
+- lib/polars/io/ipc.rb
+- lib/polars/io/json.rb
+- lib/polars/io/ndjson.rb
+- lib/polars/io/parquet.rb
 - lib/polars/lazy_frame.rb
 - lib/polars/lazy_group_by.rb
 - lib/polars/list_expr.rb
@@ -95,9 +103,8 @@ files:
 - lib/polars/testing.rb
 - lib/polars/utils.rb
 - lib/polars/version.rb
-- lib/polars/when.rb
-- lib/polars/when_then.rb
-homepage: https://github.com/ankane/polars-ruby
+- lib/polars/whenthen.rb
+homepage: https://github.com/ankane/ruby-polars
 licenses:
 - MIT
 metadata: {}

data/lib/polars/when.rb

@@ -1,16 +0,0 @@
-module Polars
-  # @private
-  class When
-    attr_accessor :_rbwhen
-
-    def initialize(rbwhen)
-      self._rbwhen = rbwhen
-    end
-
-    def then(expr)
-      expr = Utils.expr_to_lit_or_expr(expr)
-      rbwhenthen = _rbwhen._then(expr._rbexpr)
-      WhenThen.new(rbwhenthen)
-    end
-  end
-end

data/lib/polars/when_then.rb

@@ -1,19 +0,0 @@
-module Polars
-  # @private
-  class WhenThen
-    attr_accessor :_rbwhenthen
-
-    def initialize(rbwhenthen)
-      self._rbwhenthen = rbwhenthen
-    end
-
-    def when(predicate)
-      WhenThenThen.new(_rbwhenthen.when(predicate._rbexpr))
-    end
-
-    def otherwise(expr)
-      expr = Utils.expr_to_lit_or_expr(expr)
-      Utils.wrap_expr(_rbwhenthen.otherwise(expr._rbexpr))
-    end
-  end
-end