polars-df 0.1.2 → 0.1.3

data/lib/polars/expr_dispatch.rb

@@ -0,0 +1,14 @@
+module Polars
+  # @private
+  module ExprDispatch
+    private
+
+    def method_missing(method, ...)
+      return super unless self.class.method_defined?(method)
+
+      s = Utils.wrap_s(_s)
+      expr = Utils.col(s.name)
+      s.to_frame.select(expr.send(method, ...)).to_series
+    end
+  end
+end

data/lib/polars/functions.rb

@@ -1,5 +1,51 @@
 module Polars
   module Functions
+    # Convert categorical variables into dummy/indicator variables.
+    #
+    # @param df [DataFrame]
+    #   DataFrame to convert.
+    # @param columns [Array, nil]
+    #   A subset of columns to convert to dummy variables. `nil` means
+    #   "all columns".
+    #
+    # @return [DataFrame]
+    def get_dummies(df, columns: nil)
+      df.to_dummies(columns: columns)
+    end
+
+    # Aggregate multiple Dataframes/Series to a single DataFrame/Series.
+    #
+    # @param items [Object]
+    #   DataFrames/Series/LazyFrames to concatenate.
+    # @param rechunk [Boolean]
+    #   Make sure that all data is in contiguous memory.
+    # @param how ["vertical", "diagonal", "horizontal"]
+    #   Lazy only supports the 'vertical' strategy.
+    #
+    #   - Vertical: applies multiple `vstack` operations.
+    #   - Diagonal: finds a union between the column schemas and fills missing column values with null.
+    #   - Horizontal: stacks Series horizontally and fills with nulls if the lengths don't match.
+    # @param parallel [Boolean]
+    #   Only relevant for LazyFrames. This determines if the concatenated
+    #   lazy computations may be executed in parallel.
+    #
+    # @return [Object]
+    #
+    # @example
+    #   df1 = Polars::DataFrame.new({"a" => [1], "b" => [3]})
+    #   df2 = Polars::DataFrame.new({"a" => [2], "b" => [4]})
+    #   Polars.concat([df1, df2])
+    #   # =>
+    #   # shape: (2, 2)
+    #   # ┌─────┬─────┐
+    #   # │ a   ┆ b   │
+    #   # │ --- ┆ --- │
+    #   # │ i64 ┆ i64 │
+    #   # ╞═════╪═════╡
+    #   # │ 1   ┆ 3   │
+    #   # ├╌╌╌╌╌┼╌╌╌╌╌┤
+    #   # │ 2   ┆ 4   │
+    #   # └─────┴─────┘
     def concat(items, rechunk: true, how: "vertical", parallel: true)
       if items.empty?
         raise ArgumentError, "cannot concat empty list"
@@ -41,5 +87,178 @@ module Polars
         out
       end
     end
+
+    # Create a range of type `Datetime` (or `Date`).
+    #
+    # @param low [Object]
+    #   Lower bound of the date range.
+    # @param high [Object]
+    #   Upper bound of the date range.
+    # @param interval [Object]
+    #   Interval periods. It can be a polars duration string, such as `3d12h4m25s`
+    #   representing 3 days, 12 hours, 4 minutes, and 25 seconds.
+    # @param lazy [Boolean]
+    #   Return an expression.
+    # @param closed ["both", "left", "right", "none"]
+    #   Define whether the temporal window interval is closed or not.
+    # @param name [String]
+    #   Name of the output Series.
+    # @param time_unit [nil, "ns", "us", "ms"]
+    #   Set the time unit.
+    # @param time_zone [String]
+    #   Optional timezone
+    #
+    # @return [Object]
+    #
+    # @note
+    #   If both `low` and `high` are passed as date types (not datetime), and the
+    #   interval granularity is no finer than 1d, the returned range is also of
+    #   type date. All other permutations return a datetime Series.
+    #
+    # @example Using polars duration string to specify the interval
+    #   Polars.date_range(Date.new(2022, 1, 1), Date.new(2022, 3, 1), "1mo", name: "drange")
+    #   # =>
+    #   # shape: (3,)
+    #   # Series: 'drange' [date]
+    #   # [
+    #   #         2022-01-01
+    #   #         2022-02-01
+    #   #         2022-03-01
+    #   # ]
+    #
+    # @example Using `timedelta` object to specify the interval:
+    #   Polars.date_range(
+    #       DateTime.new(1985, 1, 1),
+    #       DateTime.new(1985, 1, 10),
+    #       "1d12h",
+    #       time_unit: "ms"
+    #   )
+    #   # =>
+    #   # shape: (7,)
+    #   # Series: '' [datetime[ms]]
+    #   # [
+    #   #         1985-01-01 00:00:00
+    #   #         1985-01-02 12:00:00
+    #   #         1985-01-04 00:00:00
+    #   #         1985-01-05 12:00:00
+    #   #         1985-01-07 00:00:00
+    #   #         1985-01-08 12:00:00
+    #   #         1985-01-10 00:00:00
+    #   # ]
+    def date_range(
+      low,
+      high,
+      interval,
+      lazy: false,
+      closed: "both",
+      name: nil,
+      time_unit: nil,
+      time_zone: nil
+    )
+      if defined?(ActiveSupport::Duration) && interval.is_a?(ActiveSupport::Duration)
+        raise Todo
+      else
+        interval = interval.to_s
+        if interval.include?(" ")
+          interval = interval.gsub(" ", "")
+        end
+      end
+
+      if low.is_a?(Expr) || high.is_a?(Expr) || lazy
+        low = Utils.expr_to_lit_or_expr(low, str_to_lit: true)
+        high = Utils.expr_to_lit_or_expr(high, str_to_lit: true)
+        return Utils.wrap_expr(
+          _rb_date_range_lazy(low, high, interval, closed, name, time_zone)
+        )
+      end
+
+      low, low_is_date = _ensure_datetime(low)
+      high, high_is_date = _ensure_datetime(high)
+
+      if !time_unit.nil?
+        tu = time_unit
+      elsif interval.include?("ns")
+        tu = "ns"
+      else
+        tu = "us"
+      end
+
+      start = Utils._datetime_to_pl_timestamp(low, tu)
+      stop = Utils._datetime_to_pl_timestamp(high, tu)
+      if name.nil?
+        name = ""
+      end
+
+      dt_range = Utils.wrap_s(
+        _rb_date_range(start, stop, interval, closed, name, tu, time_zone)
+      )
+      if low_is_date && high_is_date && !["h", "m", "s"].any? { |v| _interval_granularity(interval).end_with?(v) }
+        dt_range = dt_range.cast(Date)
+      end
+
+      dt_range
+    end
+
+    # def cut
+    # end
+
+    # def align_frames
+    # end
+
+    # Return a new Series of given length and type, filled with ones.
+    #
+    # @param n [Integer]
+    #   Number of elements in the `Series`
+    # @param dtype [Symbol]
+    #   DataType of the elements, defaults to `:f64`
+    #
+    # @return [Series]
+    #
+    # @note
+    #   In the lazy API you should probably not use this, but use `lit(1)`
+    #   instead.
+    def ones(n, dtype: nil)
+      s = Series.new([1.0])
+      if dtype
+        s = s.cast(dtype)
+      end
+      s.new_from_index(0, n)
+    end
+
+    # Return a new Series of given length and type, filled with zeros.
+    #
+    # @param n [Integer]
+    #   Number of elements in the `Series`
+    # @param dtype [Symbol]
+    #   DataType of the elements, defaults to `:f64`
+    #
+    # @return [Series]
+    #
+    # @note
+    #   In the lazy API you should probably not use this, but use `lit(0)`
+    #   instead.
+    def zeros(n, dtype: nil)
+      s = Series.new([0.0])
+      if dtype
+        s = s.cast(dtype)
+      end
+      s.new_from_index(0, n)
+    end
+
+    private
+
+    def _ensure_datetime(value)
+      is_date_type = false
+      if !value.is_a?(DateTime)
+        value = DateTime.new(value.year, value.month, value.day)
+        is_date_type = true
+      end
+      [value, is_date_type]
+    end
+
+    # TODO
+    def _interval_granularity(interval)
+      interval
+    end
   end
 end