polars-df 0.6.0-x86_64-linux → 0.8.0-x86_64-linux

data/lib/polars/data_types.rb

@@ -120,9 +120,12 @@ module Polars
   end
 
   # UTF-8 encoded string type.
-  class Utf8 < DataType
+  class String < DataType
   end
 
+  # Allow Utf8 as an alias for String
+  Utf8 = String
+
   # Binary type.
   class Binary < DataType
   end

data/lib/polars/date_time_expr.rb

@@ -97,15 +97,20 @@ module Polars
     #   # │ 2001-01-01 00:50:00 ┆ 2001-01-01 00:30:00 │
     #   # │ 2001-01-01 01:00:00 ┆ 2001-01-01 01:00:00 │
     #   # └─────────────────────┴─────────────────────┘
-    def truncate(every, offset: nil)
+    def truncate(every, offset: nil, use_earliest: nil)
       if offset.nil?
         offset = "0ns"
       end
 
+      if !every.is_a?(Expr)
+        every = Utils._timedelta_to_pl_duration(every)
+      end
+      every = Utils.parse_as_expression(every, str_as_lit: true)
+
       Utils.wrap_expr(
         _rbexpr.dt_truncate(
-          Utils._timedelta_to_pl_duration(every),
-          Utils._timedelta_to_pl_duration(offset)
+          every,
+          Utils._timedelta_to_pl_duration(offset),
         )
       )
     end
@@ -360,7 +365,7 @@ module Polars
     #   # ┌──────┐
     #   # │ date │
     #   # │ ---  │
-    #   # │ u32  │
+    #   # │ i8   │
     #   # ╞══════╡
     #   # │ 1    │
     #   # │ 2    │
@@ -402,7 +407,7 @@ module Polars
     #   # ┌──────┐
     #   # │ date │
     #   # │ ---  │
-    #   # │ u32  │
+    #   # │ i8   │
     #   # ╞══════╡
     #   # │ 1    │
     #   # │ 2    │
@@ -444,7 +449,7 @@ module Polars
     #   # ┌──────┐
     #   # │ date │
     #   # │ ---  │
-    #   # │ u32  │
+    #   # │ i8   │
     #   # ╞══════╡
     #   # │ 1    │
     #   # │ 5    │
@@ -491,7 +496,7 @@ module Polars
     #   # ┌─────────┬──────────────┬─────────────┐
     #   # │ weekday ┆ day_of_month ┆ day_of_year │
     #   # │ ---     ┆ ---          ┆ ---         │
-    #   # │ u32     ┆ u32          ┆ u32         │
+    #   # │ i8      ┆ i8           ┆ i16         │
     #   # ╞═════════╪══════════════╪═════════════╡
     #   # │ 1       ┆ 1            ┆ 1           │
     #   # │ 4       ┆ 4            ┆ 4           │
@@ -539,7 +544,7 @@ module Polars
     #   # ┌─────────┬──────────────┬─────────────┐
     #   # │ weekday ┆ day_of_month ┆ day_of_year │
     #   # │ ---     ┆ ---          ┆ ---         │
-    #   # │ u32     ┆ u32          ┆ u32         │
+    #   # │ i8      ┆ i8           ┆ i16         │
     #   # ╞═════════╪══════════════╪═════════════╡
     #   # │ 1       ┆ 1            ┆ 1           │
     #   # │ 4       ┆ 4            ┆ 4           │
@@ -587,7 +592,7 @@ module Polars
     #   # ┌─────────┬──────────────┬─────────────┐
     #   # │ weekday ┆ day_of_month ┆ day_of_year │
     #   # │ ---     ┆ ---          ┆ ---         │
-    #   # │ u32     ┆ u32          ┆ u32         │
+    #   # │ i8      ┆ i8           ┆ i16         │
     #   # ╞═════════╪══════════════╪═════════════╡
     #   # │ 1       ┆ 1            ┆ 1           │
     #   # │ 4       ┆ 4            ┆ 4           │
@@ -649,7 +654,7 @@ module Polars
     #   # ┌──────┐
     #   # │ date │
     #   # │ ---  │
-    #   # │ u32  │
+    #   # │ i8   │
     #   # ╞══════╡
     #   # │ 0    │
     #   # │ 12   │
@@ -690,7 +695,7 @@ module Polars
     #   # ┌──────┐
     #   # │ date │
     #   # │ ---  │
-    #   # │ u32  │
+    #   # │ i8   │
     #   # ╞══════╡
     #   # │ 0    │
     #   # │ 2    │
@@ -739,7 +744,7 @@ module Polars
     #   # ┌──────┐
     #   # │ secs │
     #   # │ ---  │
-    #   # │ u32  │
+    #   # │ i8   │
     #   # ╞══════╡
     #   # │ 0    │
     #   # │ 3    │
@@ -784,7 +789,7 @@ module Polars
     #   # ┌──────┐
     #   # │ date │
     #   # │ ---  │
-    #   # │ u32  │
+    #   # │ i8   │
     #   # ╞══════╡
     #   # │ 0    │
     #   # │ 2    │
@@ -1026,21 +1031,10 @@ module Polars
     #   Time zone for the `Datetime` Series.
     #
     # @return [Expr]
-    def replace_time_zone(tz, use_earliest: nil)
-      Utils.wrap_expr(_rbexpr.dt_replace_time_zone(tz, use_earliest))
-    end
-
-    # Localize tz-naive Datetime Series to tz-aware Datetime Series.
-    #
-    # This method takes a naive Datetime Series and makes this time zone aware.
-    # It does not move the time to another time zone.
-    #
-    # @param tz [String]
-    #   Time zone for the `Datetime` Series.
-    #
-    # @return [Expr]
-    def tz_localize(tz)
-      Utils.wrap_expr(_rbexpr.dt_tz_localize(tz))
+    def replace_time_zone(tz, use_earliest: nil, ambiguous: "raise")
+      ambiguous = Utils.rename_use_earliest_to_ambiguous(use_earliest, ambiguous)
+      ambiguous = Polars.lit(ambiguous) unless ambiguous.is_a?(Expr)
+      Utils.wrap_expr(_rbexpr.dt_replace_time_zone(tz, ambiguous._rbexpr))
     end
 
     # Extract the days from a Duration type.
@@ -1348,6 +1342,7 @@ module Polars
     #   # │ 2006-01-01 00:00:00 ┆ 2003-11-01 00:00:00 │
     #   # └─────────────────────┴─────────────────────┘
     def offset_by(by)
+      by = Utils.parse_as_expression(by, str_as_lit: true)
       Utils.wrap_expr(_rbexpr.dt_offset_by(by))
     end
 

data/lib/polars/date_time_name_space.rb

@@ -23,18 +23,8 @@ module Polars
     # @return [Object]
     #
     # @example
-    #   date = Polars.date_range(DateTime.new(2001, 1, 1), DateTime.new(2001, 1, 3), "1d")
-    #   # =>
-    #   # shape: (3,)
-    #   # Series: '' [datetime[μs]]
-    #   # [
-    #   #         2001-01-01 00:00:00
-    #   #         2001-01-02 00:00:00
-    #   #         2001-01-03 00:00:00
-    #   # ]
-    #
-    # @example
-    #   date.dt.min
+    #   s = Polars.date_range(DateTime.new(2001, 1, 1), DateTime.new(2001, 1, 3), "1d")
+    #   s.dt.min
     #   # => 2001-01-01 00:00:00 UTC
     def min
       Utils.wrap_s(_s).min
@@ -45,18 +35,8 @@ module Polars
     # @return [Object]
     #
     # @example
-    #   date = Polars.date_range(DateTime.new(2001, 1, 1), DateTime.new(2001, 1, 3), "1d")
-    #   # =>
-    #   # shape: (3,)
-    #   # Series: '' [datetime[μs]]
-    #   # [
-    #   #         2001-01-01 00:00:00
-    #   #         2001-01-02 00:00:00
-    #   #         2001-01-03 00:00:00
-    #   # ]
-    #
-    # @example
-    #   date.dt.max
+    #   s = Polars.date_range(DateTime.new(2001, 1, 1), DateTime.new(2001, 1, 3), "1d")
+    #   s.dt.max
     #   # => 2001-01-03 00:00:00 UTC
     def max
       Utils.wrap_s(_s).max
@@ -240,7 +220,7 @@ module Polars
     #   date.dt.quarter
     #   # =>
     #   # shape: (4,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i8]
     #   # [
     #   #         1
     #   #         1
@@ -278,7 +258,7 @@ module Polars
     #   date.dt.month
     #   # =>
     #   # shape: (4,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i8]
     #   # [
     #   #         1
     #   #         2
@@ -316,7 +296,7 @@ module Polars
     #   date.dt.week
     #   # =>
     #   # shape: (4,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i8]
     #   # [
     #   #         1
     #   #         5
@@ -356,7 +336,7 @@ module Polars
     #   date.dt.weekday
     #   # =>
     #   # shape: (7,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i8]
     #   # [
     #   #         1
     #   #         2
@@ -398,7 +378,7 @@ module Polars
     #   date.dt.day
     #   # =>
     #   # shape: (5,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i8]
     #   # [
     #   #         1
     #   #         3
@@ -436,7 +416,7 @@ module Polars
     #   date.dt.ordinal_day
     #   # =>
     #   # shape: (3,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i16]
     #   # [
     #   #         1
     #   #         32
@@ -472,7 +452,7 @@ module Polars
     #   date.dt.hour
     #   # =>
     #   # shape: (4,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i8]
     #   # [
     #   #         0
     #   #         1
@@ -508,7 +488,7 @@ module Polars
     #   date.dt.minute
     #   # =>
     #   # shape: (3,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i8]
     #   # [
     #   #         0
     #   #         2
@@ -551,7 +531,7 @@ module Polars
     #   date.dt.second
     #   # =>
     #   # shape: (9,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i8]
     #   # [
     #   #         0
     #   #         0
@@ -613,7 +593,7 @@ module Polars
     #   date.dt.millisecond
     #   # =>
     #   # shape: (9,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i32]
     #   # [
     #   #         0
     #   #         500
@@ -658,7 +638,7 @@ module Polars
     #   date.dt.microsecond
     #   # =>
     #   # shape: (9,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i32]
     #   # [
     #   #         0
     #   #         500000
@@ -703,7 +683,7 @@ module Polars
     #   date.dt.nanosecond
     #   # =>
     #   # shape: (9,)
-    #   # Series: '' [u32]
+    #   # Series: '' [i32]
     #   # [
     #   #         0
     #   #         500000000
@@ -1400,7 +1380,7 @@ module Polars
     #   #         2001-01-01 00:30:00
     #   #         2001-01-01 01:00:00
     #   # ]
-    def truncate(every, offset: nil)
+    def truncate(every, offset: nil, use_earliest: nil)
       super
     end
 

data/lib/polars/dynamic_group_by.rb

@@ -2,7 +2,7 @@ module Polars
   # A dynamic grouper.
   #
   # This has an `.agg` method which allows you to run all polars expressions in a
-  # groupby context.
+  # group by context.
   class DynamicGroupBy
     def initialize(
       df,
@@ -34,7 +34,7 @@ module Polars
 
     def agg(aggs)
       @df.lazy
-        .groupby_dynamic(
+        .group_by_dynamic(
           @time_column,
           every: @every,
           period: @period,