polars-df 0.23.0 → 0.24.0

data/lib/polars.rb

@@ -26,8 +26,8 @@ require_relative "polars/catalog/unity/namespace_info"
 require_relative "polars/catalog/unity/table_info"
 require_relative "polars/config"
 require_relative "polars/convert"
-require_relative "polars/plot"
 require_relative "polars/data_frame"
+require_relative "polars/data_frame_plot"
 require_relative "polars/data_types"
 require_relative "polars/data_type_expr"
 require_relative "polars/data_type_group"
@@ -41,6 +41,7 @@ require_relative "polars/functions/business"
 require_relative "polars/functions/col"
 require_relative "polars/functions/datatype"
 require_relative "polars/functions/eager"
+require_relative "polars/functions/escape_regex"
 require_relative "polars/functions/lazy"
 require_relative "polars/functions/len"
 require_relative "polars/functions/lit"
@@ -52,10 +53,13 @@ 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/linear_space"
 require_relative "polars/functions/range/time_range"
 require_relative "polars/group_by"
 require_relative "polars/iceberg_dataset"
+require_relative "polars/in_process_query"
 require_relative "polars/io/avro"
+require_relative "polars/io/cloud"
 require_relative "polars/io/csv"
 require_relative "polars/io/database"
 require_relative "polars/io/delta"
@@ -79,6 +83,7 @@ require_relative "polars/schema"
 require_relative "polars/selector"
 require_relative "polars/selectors"
 require_relative "polars/series"
+require_relative "polars/series_plot"
 require_relative "polars/slice"
 require_relative "polars/sql_context"
 require_relative "polars/string_cache"
@@ -89,9 +94,14 @@ require_relative "polars/struct_name_space"
 require_relative "polars/testing"
 require_relative "polars/utils"
 require_relative "polars/utils/constants"
+require_relative "polars/utils/construction/data_frame"
+require_relative "polars/utils/construction/series"
+require_relative "polars/utils/construction/utils"
 require_relative "polars/utils/convert"
+require_relative "polars/utils/deprecation"
 require_relative "polars/utils/parse"
 require_relative "polars/utils/serde"
+require_relative "polars/utils/unstable"
 require_relative "polars/utils/various"
 require_relative "polars/utils/wrap"
 require_relative "polars/version"
@@ -105,6 +115,9 @@ module Polars
   # @private
   F = self
 
+  # @private
+  NO_DEFAULT = Object.new
+
   # @private
   N_INFER_DEFAULT = 100
 
@@ -121,4 +134,37 @@ module Polars
   def self.thread_pool_size
     Plr.thread_pool_size
   end
+
+  # Return the data type used for Polars indexing.
+  #
+  # @return [Object]
+  #
+  # @example
+  #   Polars.get_index_type
+  #   # => Polars::UInt32
+  def self.get_index_type
+    Plr.get_index_type
+  end
+
+  # Return detailed Polars build information.
+  #
+  # @return [Hash]
+  #
+  # @example
+  #   Polars.build_info
+  def self.build_info
+    {"version" => VERSION}
+  end
+
+  # Print out the version of Polars and its optional dependencies.
+  #
+  # @return [nil]
+  def self.show_versions
+    puts "--------Version info---------"
+    puts "Polars: #{VERSION}"
+    puts "Index type: #{get_index_type}"
+    puts "Platform: #{RUBY_PLATFORM}"
+    puts "Ruby: #{RUBY_VERSION}"
+    nil
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: polars-df
 version: !ruby/object:Gem::Version
-  version: 0.23.0
+  version: 0.24.0
 platform: ruby
 authors:
 - Andrew Kane
@@ -51,8 +51,9 @@ files:
 - README.md
 - ext/polars/Cargo.toml
 - ext/polars/extconf.rb
-- ext/polars/src/allocator.rs
 - ext/polars/src/batched_csv.rs
+- ext/polars/src/c_api/allocator.rs
+- ext/polars/src/c_api/mod.rs
 - ext/polars/src/catalog/mod.rs
 - ext/polars/src/catalog/unity.rs
 - ext/polars/src/conversion/any_value.rs
@@ -96,15 +97,19 @@ files:
 - ext/polars/src/functions/random.rs
 - ext/polars/src/functions/range.rs
 - ext/polars/src/functions/string_cache.rs
+- ext/polars/src/functions/strings.rs
+- ext/polars/src/functions/utils.rs
 - ext/polars/src/functions/whenthen.rs
 - ext/polars/src/interop/arrow/mod.rs
-- ext/polars/src/interop/arrow/to_ruby.rs
+- ext/polars/src/interop/arrow/to_rb.rs
+- ext/polars/src/interop/arrow/to_rust.rs
 - ext/polars/src/interop/mod.rs
 - ext/polars/src/interop/numo/mod.rs
 - ext/polars/src/interop/numo/numo_rs.rs
 - ext/polars/src/interop/numo/to_numo_df.rs
 - ext/polars/src/interop/numo/to_numo_series.rs
 - ext/polars/src/io/mod.rs
+- ext/polars/src/lazyframe/exitable.rs
 - ext/polars/src/lazyframe/general.rs
 - ext/polars/src/lazyframe/mod.rs
 - ext/polars/src/lazyframe/optflags.rs
@@ -131,6 +136,10 @@ files:
 - ext/polars/src/series/mod.rs
 - ext/polars/src/series/scatter.rs
 - ext/polars/src/sql.rs
+- ext/polars/src/testing/frame.rs
+- ext/polars/src/testing/mod.rs
+- ext/polars/src/testing/series.rs
+- ext/polars/src/timeout.rs
 - ext/polars/src/utils.rs
 - lib/polars-df.rb
 - lib/polars.rb
@@ -149,6 +158,7 @@ files:
 - lib/polars/config.rb
 - lib/polars/convert.rb
 - lib/polars/data_frame.rb
+- lib/polars/data_frame_plot.rb
 - lib/polars/data_type_expr.rb
 - lib/polars/data_type_group.rb
 - lib/polars/data_types.rb
@@ -165,6 +175,7 @@ files:
 - lib/polars/functions/col.rb
 - lib/polars/functions/datatype.rb
 - lib/polars/functions/eager.rb
+- lib/polars/functions/escape_regex.rb
 - lib/polars/functions/lazy.rb
 - lib/polars/functions/len.rb
 - lib/polars/functions/lit.rb
@@ -172,12 +183,15 @@ files:
 - 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/linear_space.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/iceberg_dataset.rb
+- lib/polars/in_process_query.rb
 - lib/polars/io/avro.rb
+- lib/polars/io/cloud.rb
 - lib/polars/io/csv.rb
 - lib/polars/io/database.rb
 - lib/polars/io/delta.rb
@@ -194,7 +208,6 @@ files:
 - lib/polars/list_name_space.rb
 - lib/polars/meta_expr.rb
 - lib/polars/name_expr.rb
-- lib/polars/plot.rb
 - lib/polars/query_opt_flags.rb
 - lib/polars/rolling_group_by.rb
 - lib/polars/scan_cast_options.rb
@@ -202,6 +215,7 @@ files:
 - lib/polars/selector.rb
 - lib/polars/selectors.rb
 - lib/polars/series.rb
+- lib/polars/series_plot.rb
 - lib/polars/slice.rb
 - lib/polars/sql_context.rb
 - lib/polars/string_cache.rb
@@ -212,9 +226,14 @@ files:
 - lib/polars/testing.rb
 - lib/polars/utils.rb
 - lib/polars/utils/constants.rb
+- lib/polars/utils/construction/data_frame.rb
+- lib/polars/utils/construction/series.rb
+- lib/polars/utils/construction/utils.rb
 - lib/polars/utils/convert.rb
+- lib/polars/utils/deprecation.rb
 - lib/polars/utils/parse.rb
 - lib/polars/utils/serde.rb
+- lib/polars/utils/unstable.rb
 - lib/polars/utils/various.rb
 - lib/polars/utils/wrap.rb
 - lib/polars/version.rb
@@ -230,14 +249,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '3.2'
+      version: '3.3'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.9
+rubygems_version: 4.0.3
 specification_version: 4
 summary: Blazingly fast DataFrames for Ruby
 test_files: []

data/ext/polars/src/allocator.rs

@@ -1,13 +0,0 @@
-#[cfg(target_os = "linux")]
-use jemallocator::Jemalloc;
-
-#[cfg(not(any(target_os = "linux", target_os = "windows")))]
-use mimalloc::MiMalloc;
-
-#[global_allocator]
-#[cfg(target_os = "linux")]
-static ALLOC: Jemalloc = Jemalloc;
-
-#[global_allocator]
-#[cfg(not(any(target_os = "linux", target_os = "windows")))]
-static ALLOC: MiMalloc = MiMalloc;

data/lib/polars/plot.rb

@@ -1,109 +0,0 @@
-module Polars
-  module Plot
-    # Plot data.
-    #
-    # @return [Vega::LiteChart]
-    def plot(x = nil, y = nil, type: nil, group: nil, stacked: nil)
-      require "vega"
-
-      raise ArgumentError, "Must specify columns" if columns.size != 2 && (!x || !y)
-      x ||= columns[0]
-      y ||= columns[1]
-      type ||= begin
-        if self[x].numeric? && self[y].numeric?
-          "scatter"
-        elsif self[x].utf8? && self[y].numeric?
-          "column"
-        elsif (self[x].dtype == Date || self[x].dtype.is_a?(Datetime)) && self[y].numeric?
-          "line"
-        else
-          raise "Cannot determine type. Use the type option."
-        end
-      end
-      df = self[(group.nil? ? [x, y] : [x, y, group]).map(&:to_s).uniq]
-      data = df.rows(named: true)
-
-      case type
-      when "line", "area"
-        x_type =
-          if df[x].numeric?
-            "quantitative"
-          elsif df[x].datelike?
-            "temporal"
-          else
-            "nominal"
-          end
-
-        scale = x_type == "temporal" ? {type: "utc"} : {}
-        encoding = {
-          x: {field: x, type: x_type, scale: scale},
-          y: {field: y, type: "quantitative"}
-        }
-        encoding[:color] = {field: group} if group
-
-        Vega.lite
-          .data(data)
-          .mark(type: type, tooltip: true, interpolate: "cardinal", point: {size: 60})
-          .encoding(encoding)
-          .config(axis: {labelFontSize: 12})
-      when "pie"
-        raise ArgumentError, "Cannot use group option with pie chart" unless group.nil?
-
-        Vega.lite
-          .data(data)
-          .mark(type: "arc", tooltip: true)
-          .encoding(
-            color: {field: x, type: "nominal", sort: "none", axis: {title: nil}, legend: {labelFontSize: 12}},
-            theta: {field: y, type: "quantitative"}
-          )
-          .view(stroke: nil)
-      when "column"
-        encoding = {
-          x: {field: x, type: "nominal", sort: "none", axis: {labelAngle: 0}},
-          y: {field: y, type: "quantitative"}
-        }
-        if group
-          encoding[:color] = {field: group}
-          encoding[:xOffset] = {field: group} unless stacked
-        end
-
-        Vega.lite
-          .data(data)
-          .mark(type: "bar", tooltip: true)
-          .encoding(encoding)
-          .config(axis: {labelFontSize: 12})
-      when "bar"
-        encoding = {
-          # TODO determine label angle
-          y: {field: x, type: "nominal", sort: "none", axis: {labelAngle: 0}},
-          x: {field: y, type: "quantitative"}
-        }
-        if group
-          encoding[:color] = {field: group}
-          encoding[:yOffset] = {field: group} unless stacked
-        end
-
-        Vega.lite
-          .data(data)
-          .mark(type: "bar", tooltip: true)
-          .encoding(encoding)
-          .config(axis: {labelFontSize: 12})
-      when "scatter"
-        encoding = {
-          x: {field: x, type: "quantitative", scale: {zero: false}},
-          y: {field: y, type: "quantitative", scale: {zero: false}},
-          size: {value: 60}
-        }
-        encoding[:color] = {field: group} if group
-
-        Vega.lite
-          .data(data)
-          .mark(type: "circle", tooltip: true)
-          .encoding(encoding)
-          .config(axis: {labelFontSize: 12})
-      else
-        raise ArgumentError, "Invalid type: #{type}"
-      end
-    end
-  end
-end