polars-df 0.2.4-x86_64-linux → 0.3.0-x86_64-linux

data/lib/polars/lazy_functions.rb

@@ -15,7 +15,7 @@ module Polars
       if name.is_a?(DataType)
         Utils.wrap_expr(_dtype_cols([name]))
       elsif name.is_a?(Array)
-        if name.length == 0 || name[0].is_a?(String) || name[0].is_a?(Symbol)
+        if name.length == 0 || Utils.strlike?(name[0])
           name = name.map { |v| v.is_a?(Symbol) ? v.to_s : v }
           Utils.wrap_expr(RbExpr.cols(name))
         elsif Utils.is_polars_dtype(name[0])
@@ -46,9 +46,7 @@ module Polars
     #   # │ i64 ┆ i64 ┆ list[f32]  │
     #   # ╞═════╪═════╪════════════╡
     #   # │ 1   ┆ 4   ┆ [1.0, 2.0] │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 8   ┆ 5   ┆ [2.0, 1.0] │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 3   ┆ 2   ┆ [2.0, 1.0] │
     #   # └─────┴─────┴────────────┘
     def element
@@ -119,7 +117,7 @@ module Polars
     def max(column)
       if column.is_a?(Series)
         column.max
-      elsif column.is_a?(String) || column.is_a?(Symbol)
+      elsif Utils.strlike?(column)
         col(column).max
       else
         exprs = Utils.selection_to_rbexpr_list(column)
@@ -141,7 +139,7 @@ module Polars
     def min(column)
       if column.is_a?(Series)
         column.min
-      elsif column.is_a?(String) || column.is_a?(Symbol)
+      elsif Utils.strlike?(column)
         col(column).min
       else
         exprs = Utils.selection_to_rbexpr_list(column)
@@ -156,7 +154,7 @@ module Polars
     def sum(column)
       if column.is_a?(Series)
         column.sum
-      elsif column.is_a?(String) || column.is_a?(Symbol)
+      elsif Utils.strlike?(column)
         col(column.to_s).sum
       elsif column.is_a?(Array)
         exprs = Utils.selection_to_rbexpr_list(column)
@@ -322,7 +320,6 @@ module Polars
     #   # │ i64 ┆ i64 ┆ i64 │
     #   # ╞═════╪═════╪═════╡
     #   # │ 1   ┆ 3   ┆ 5   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ 2   ┆ 4   ┆ 6   │
     #   # └─────┴─────┴─────┘
     #
@@ -336,7 +333,6 @@ module Polars
     #   # │ i64 │
     #   # ╞═════╡
     #   # │ 1   │
-    #   # ├╌╌╌╌╌┤
     #   # │ 3   │
     #   # └─────┘
     #
@@ -350,13 +346,12 @@ module Polars
     #   # │ i64 ┆ i64 ┆ i64 ┆ struct[2] │
     #   # ╞═════╪═════╪═════╪═══════════╡
     #   # │ 1   ┆ 3   ┆ 5   ┆ {1,6}     │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 2   ┆ 4   ┆ 6   ┆ {2,8}     │
     #   # └─────┴─────┴─────┴───────────┘
     def cumsum(column)
       if column.is_a?(Series)
         column.cumsum
-      elsif column.is_a?(String)
+      elsif Utils.strlike?(column)
         col(column).cumsum
       else
         cumfold(lit(0).cast(:u32), ->(a, b) { a + b }, column).alias("cumsum")
@@ -380,10 +375,10 @@ module Polars
     #
     # @return [Expr]
     def spearman_rank_corr(a, b, ddof: 1, propagate_nans: false)
-      if a.is_a?(String)
+      if Utils.strlike?(a)
         a = col(a)
       end
-      if b.is_a?(String)
+      if Utils.strlike?(b)
         b = col(b)
       end
       Utils.wrap_expr(RbExpr.spearman_rank_corr(a._rbexpr, b._rbexpr, ddof, propagate_nans))
@@ -400,10 +395,10 @@ module Polars
     #
     # @return [Expr]
     def pearson_corr(a, b, ddof: 1)
-      if a.is_a?(String)
+      if Utils.strlike?(a)
         a = col(a)
       end
-      if b.is_a?(String)
+      if Utils.strlike?(b)
         b = col(b)
       end
       Utils.wrap_expr(RbExpr.pearson_corr(a._rbexpr, b._rbexpr, ddof))
@@ -418,10 +413,10 @@ module Polars
     #
     # @return [Expr]
     def cov(a, b)
-      if a.is_a?(String)
+      if Utils.strlike?(a)
         a = col(a)
       end
-      if b.is_a?(String)
+      if Utils.strlike?(b)
         b = col(b)
       end
       Utils.wrap_expr(RbExpr.cov(a._rbexpr, b._rbexpr))
@@ -486,7 +481,7 @@ module Polars
     #
     # @return [Expr]
     def any(name)
-      if name.is_a?(String)
+      if Utils.strlike?(name)
         col(name).any
       else
         fold(lit(false), ->(a, b) { a.cast(:bool) | b.cast(:bool) }, name).alias("any")
@@ -521,9 +516,7 @@ module Polars
     #   # │ i64 ┆ str  ┆ f64  │
     #   # ╞═════╪══════╪══════╡
     #   # │ 1   ┆ a    ┆ null │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ 2   ┆ b    ┆ 2.5  │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ 3   ┆ null ┆ 1.5  │
     #   # └─────┴──────┴──────┘
     #
@@ -537,9 +530,7 @@ module Polars
     #   # │ i64 ┆ f64  │
     #   # ╞═════╪══════╡
     #   # │ 1   ┆ null │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ 2   ┆ 2.5  │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ 3   ┆ 1.5  │
     #   # └─────┴──────┘
     #
@@ -553,9 +544,7 @@ module Polars
     #   # │ f64  │
     #   # ╞══════╡
     #   # │ null │
-    #   # ├╌╌╌╌╌╌┤
     #   # │ 2.5  │
-    #   # ├╌╌╌╌╌╌┤
     #   # │ 1.5  │
     #   # └──────┘
     def exclude(columns)
@@ -589,7 +578,7 @@ module Polars
     def all(name = nil)
       if name.nil?
         col("*")
-      elsif name.is_a?(String) || name.is_a?(Symbol)
+      elsif Utils.strlike?(name)
         col(name).all
       else
         raise Todo
@@ -709,7 +698,6 @@ module Polars
     #   # │ datetime[ns]        ┆ datetime[ns]        ┆ datetime[ns]        ┆ datetime[ns]            ┆ datetime[ns]        │
     #   # ╞═════════════════════╪═════════════════════╪═════════════════════╪═════════════════════════╪═════════════════════╡
     #   # │ 2022-01-08 00:00:00 ┆ 2022-01-02 00:00:00 ┆ 2022-01-01 00:00:01 ┆ 2022-01-01 00:00:00.001 ┆ 2022-01-01 01:00:00 │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 2022-01-16 00:00:00 ┆ 2022-01-04 00:00:00 ┆ 2022-01-02 00:00:02 ┆ 2022-01-02 00:00:00.002 ┆ 2022-01-02 02:00:00 │
     #   # └─────────────────────┴─────────────────────┴─────────────────────┴─────────────────────────┴─────────────────────┘
     def duration(
@@ -798,9 +786,7 @@ module Polars
     #   # │ i64 ┆ str  ┆ str  ┆ str           │
     #   # ╞═════╪══════╪══════╪═══════════════╡
     #   # │ 1   ┆ dogs ┆ play ┆ 2 dogs play   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 2   ┆ cats ┆ swim ┆ 4 cats swim   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 3   ┆ null ┆ walk ┆ null          │
     #   # └─────┴──────┴──────┴───────────────┘
     def concat_str(exprs, sep: "")
@@ -838,9 +824,7 @@ module Polars
     #   # │ str         │
     #   # ╞═════════════╡
     #   # │ foo_a_bar_1 │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ foo_b_bar_2 │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ foo_c_bar_3 │
     #   # └─────────────┘
     def format(fstring, *args)
@@ -972,7 +956,6 @@ module Polars
     #   # │ struct[4]           │
     #   # ╞═════════════════════╡
     #   # │ {1,"a",true,[1, 2]} │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ {2,"b",null,[3]}    │
     #   # └─────────────────────┘
     #
@@ -989,11 +972,8 @@ module Polars
     #   # │ i64 ┆ str   ┆ i64 ┆ struct[2]   │
     #   # ╞═════╪═══════╪═════╪═════════════╡
     #   # │ 1   ┆ one   ┆ 9   ┆ {1,"one"}   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 2   ┆ two   ┆ 8   ┆ {2,"two"}   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 3   ┆ three ┆ 7   ┆ {3,"three"} │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 4   ┆ four  ┆ 6   ┆ {4,"four"}  │
     #   # └─────┴───────┴─────┴─────────────┘
     def struct(exprs, eager: false)
@@ -1092,11 +1072,8 @@ module Polars
     #   # │ f64  ┆ f64  ┆ f64  ┆ f64  │
     #   # ╞══════╪══════╪══════╪══════╡
     #   # │ null ┆ 1.0  ┆ 1.0  ┆ 1.0  │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ null ┆ 2.0  ┆ 2.0  ┆ 2.0  │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ null ┆ null ┆ 3.0  ┆ 3.0  │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ null ┆ null ┆ null ┆ 99.9 │
     #   # └──────┴──────┴──────┴──────┘
     def coalesce(exprs)
@@ -1133,11 +1110,10 @@ module Polars
     #   # │ datetime[μs]        │
     #   # ╞═════════════════════╡
     #   # │ 2022-10-25 07:31:17 │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 2022-10-25 07:31:39 │
     #   # └─────────────────────┘
     def from_epoch(column, unit: "s", eager: false)
-      if column.is_a?(String)
+      if Utils.strlike?(column)
         column = col(column)
       elsif !column.is_a?(Series) && !column.is_a?(Expr)
         column = Series.new(column)
@@ -1181,9 +1157,7 @@ module Polars
     #   # │ i64 ┆ i64 ┆ i32     │
     #   # ╞═════╪═════╪═════════╡
     #   # │ 1   ┆ 3   ┆ -1      │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌┤
     #   # │ 3   ┆ 4   ┆ 1       │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌┤
     #   # │ 4   ┆ 0   ┆ 1       │
     #   # └─────┴─────┴─────────┘
     def when(expr)

data/lib/polars/lazy_group_by.rb

@@ -38,13 +38,9 @@ module Polars
     #   # │ str     ┆ i64 │
     #   # ╞═════════╪═════╡
     #   # │ a       ┆ 3   │
-    #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ a       ┆ 5   │
-    #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ b       ┆ 6   │
-    #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ c       ┆ 1   │
-    #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ c       ┆ 2   │
     #   # └─────────┴─────┘
     def head(n = 5)
@@ -74,13 +70,9 @@ module Polars
     #   # │ str     ┆ i64 │
     #   # ╞═════════╪═════╡
     #   # │ a       ┆ 3   │
-    #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ a       ┆ 5   │
-    #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ b       ┆ 6   │
-    #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ c       ┆ 2   │
-    #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ c       ┆ 4   │
     #   # └─────────┴─────┘
     def tail(n = 5)

data/lib/polars/list_expr.rb

@@ -24,7 +24,6 @@ module Polars
     #   # │ u32 │
     #   # ╞═════╡
     #   # │ 2   │
-    #   # ├╌╌╌╌╌┤
     #   # │ 1   │
     #   # └─────┘
     def lengths
@@ -46,7 +45,6 @@ module Polars
     #   # │ i64    │
     #   # ╞════════╡
     #   # │ 1      │
-    #   # ├╌╌╌╌╌╌╌╌┤
     #   # │ 5      │
     #   # └────────┘
     def sum
@@ -68,7 +66,6 @@ module Polars
     #   # │ i64    │
     #   # ╞════════╡
     #   # │ 1      │
-    #   # ├╌╌╌╌╌╌╌╌┤
     #   # │ 3      │
     #   # └────────┘
     def max
@@ -90,7 +87,6 @@ module Polars
     #   # │ i64    │
     #   # ╞════════╡
     #   # │ 1      │
-    #   # ├╌╌╌╌╌╌╌╌┤
     #   # │ 2      │
     #   # └────────┘
     def min
@@ -112,7 +108,6 @@ module Polars
     #   # │ f64    │
     #   # ╞════════╡
     #   # │ 1.0    │
-    #   # ├╌╌╌╌╌╌╌╌┤
     #   # │ 2.5    │
     #   # └────────┘
     def mean
@@ -138,7 +133,6 @@ module Polars
     #   # │ list[i64] │
     #   # ╞═══════════╡
     #   # │ [1, 2, 3] │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ [1, 2, 9] │
     #   # └───────────┘
     def sort(reverse: false)
@@ -164,7 +158,6 @@ module Polars
     #   # │ list[i64] │
     #   # ╞═══════════╡
     #   # │ [1, 2, 3] │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ [2, 1, 9] │
     #   # └───────────┘
     def reverse
@@ -218,7 +211,6 @@ module Polars
     #   # │ list[str]       │
     #   # ╞═════════════════╡
     #   # │ ["a", "b", "c"] │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ ["x", "y", "z"] │
     #   # └─────────────────┘
     def concat(other)
@@ -258,9 +250,7 @@ module Polars
     #   # │ i64  │
     #   # ╞══════╡
     #   # │ 3    │
-    #   # ├╌╌╌╌╌╌┤
     #   # │ null │
-    #   # ├╌╌╌╌╌╌┤
     #   # │ 1    │
     #   # └──────┘
     def get(index)
@@ -290,9 +280,7 @@ module Polars
     #   # │ i64  │
     #   # ╞══════╡
     #   # │ 3    │
-    #   # ├╌╌╌╌╌╌┤
     #   # │ null │
-    #   # ├╌╌╌╌╌╌┤
     #   # │ 1    │
     #   # └──────┘
     def first
@@ -314,9 +302,7 @@ module Polars
     #   # │ i64  │
     #   # ╞══════╡
     #   # │ 1    │
-    #   # ├╌╌╌╌╌╌┤
     #   # │ null │
-    #   # ├╌╌╌╌╌╌┤
     #   # │ 2    │
     #   # └──────┘
     def last
@@ -341,9 +327,7 @@ module Polars
     #   # │ bool  │
     #   # ╞═══════╡
     #   # │ true  │
-    #   # ├╌╌╌╌╌╌╌┤
     #   # │ false │
-    #   # ├╌╌╌╌╌╌╌┤
     #   # │ true  │
     #   # └───────┘
     def contains(item)
@@ -370,7 +354,6 @@ module Polars
     #   # │ str   │
     #   # ╞═══════╡
     #   # │ a b c │
-    #   # ├╌╌╌╌╌╌╌┤
     #   # │ x y   │
     #   # └───────┘
     def join(separator)
@@ -396,7 +379,6 @@ module Polars
     #   # │ u32 │
     #   # ╞═════╡
     #   # │ 0   │
-    #   # ├╌╌╌╌╌┤
     #   # │ 1   │
     #   # └─────┘
     def arg_min
@@ -422,7 +404,6 @@ module Polars
     #   # │ u32 │
     #   # ╞═════╡
     #   # │ 1   │
-    #   # ├╌╌╌╌╌┤
     #   # │ 0   │
     #   # └─────┘
     def arg_max
@@ -443,7 +424,7 @@ module Polars
     #   s.arr.diff
     #   # =>
     #   # shape: (2,)
-    #   # Series: 'a' [list]
+    #   # Series: 'a' [list[i64]]
     #   # [
     #   #         [null, 1, ... 1]
     #   #         [null, -8, -1]
@@ -464,7 +445,7 @@ module Polars
     #   s.arr.shift
     #   # =>
     #   # shape: (2,)
-    #   # Series: 'a' [list]
+    #   # Series: 'a' [list[i64]]
     #   # [
     #   #         [null, 1, ... 3]
     #   #         [null, 10, 2]
@@ -488,7 +469,7 @@ module Polars
     #   s.arr.slice(1, 2)
     #   # =>
     #   # shape: (2,)
-    #   # Series: 'a' [list]
+    #   # Series: 'a' [list[i64]]
     #   # [
     #   #         [2, 3]
     #   #         [2, 1]
@@ -511,7 +492,7 @@ module Polars
     #   s.arr.head(2)
     #   # =>
     #   # shape: (2,)
-    #   # Series: 'a' [list]
+    #   # Series: 'a' [list[i64]]
     #   # [
     #   #         [1, 2]
     #   #         [10, 2]
@@ -532,7 +513,7 @@ module Polars
     #   s.arr.tail(2)
     #   # =>
     #   # shape: (2,)
-    #   # Series: 'a' [list]
+    #   # Series: 'a' [list[i64]]
     #   # [
     #   #         [3, 4]
     #   #         [2, 1]
@@ -563,7 +544,6 @@ module Polars
     #   # │ struct[3]  │
     #   # ╞════════════╡
     #   # │ {1,2,3}    │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ {1,2,null} │
     #   # └────────────┘
     def to_struct(n_field_strategy: "first_non_null", name_generator: nil)
@@ -598,9 +578,7 @@ module Polars
     #   # │ i64 ┆ i64 ┆ list[f32]  │
     #   # ╞═════╪═════╪════════════╡
     #   # │ 1   ┆ 4   ┆ [1.0, 2.0] │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 8   ┆ 5   ┆ [2.0, 1.0] │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 3   ┆ 2   ┆ [2.0, 1.0] │
     #   # └─────┴─────┴────────────┘
     def eval(expr, parallel: false)

data/lib/polars/list_name_space.rb

@@ -183,7 +183,7 @@ module Polars
     #   s.arr.diff
     #   # =>
     #   # shape: (2,)
-    #   # Series: 'a' [list]
+    #   # Series: 'a' [list[i64]]
     #   # [
     #   #         [null, 1, ... 1]
     #   #         [null, -8, -1]
@@ -204,7 +204,7 @@ module Polars
     #   s.arr.shift
     #   # =>
     #   # shape: (2,)
-    #   # Series: 'a' [list]
+    #   # Series: 'a' [list[i64]]
     #   # [
     #   #         [null, 1, ... 3]
     #   #         [null, 10, 2]
@@ -228,7 +228,7 @@ module Polars
     #   s.arr.slice(1, 2)
     #   # =>
     #   # shape: (2,)
-    #   # Series: 'a' [list]
+    #   # Series: 'a' [list[i64]]
     #   # [
     #   #         [2, 3]
     #   #         [2, 1]
@@ -249,7 +249,7 @@ module Polars
     #   s.arr.head(2)
     #   # =>
     #   # shape: (2,)
-    #   # Series: 'a' [list]
+    #   # Series: 'a' [list[i64]]
     #   # [
     #   #         [1, 2]
     #   #         [10, 2]
@@ -270,7 +270,7 @@ module Polars
     #   s.arr.tail(2)
     #   # =>
     #   # shape: (2,)
-    #   # Series: 'a' [list]
+    #   # Series: 'a' [list[i64]]
     #   # [
     #   #         [3, 4]
     #   #         [2, 1]
@@ -300,7 +300,6 @@ module Polars
     #   # │ struct[3]  │
     #   # ╞════════════╡
     #   # │ {1,2,3}    │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ {1,2,null} │
     #   # └────────────┘
     def to_struct(n_field_strategy: "first_non_null", name_generator: nil)
@@ -334,9 +333,7 @@ module Polars
     #   # │ i64 ┆ i64 ┆ list[f32]  │
     #   # ╞═════╪═════╪════════════╡
     #   # │ 1   ┆ 4   ┆ [1.0, 2.0] │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 8   ┆ 5   ┆ [2.0, 1.0] │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌┤
     #   # │ 3   ┆ 2   ┆ [2.0, 1.0] │
     #   # └─────┴─────┴────────────┘
     def eval(expr, parallel: false)

data/lib/polars/series.rb

@@ -259,6 +259,17 @@ module Polars
       0 - self
     end
 
+    # Returns an enumerator.
+    #
+    # @return [Object]
+    def each
+      return to_enum(:each) unless block_given?
+
+      length.times do |i|
+        yield self[i]
+      end
+    end
+
     # Returns elements of the Series.
     #
     # @return [Object]
@@ -432,15 +443,10 @@ module Polars
     #   # │ str        ┆ f64      │
     #   # ╞════════════╪══════════╡
     #   # │ min        ┆ 1.0      │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┤
     #   # │ max        ┆ 5.0      │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┤
     #   # │ null_count ┆ 0.0      │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┤
     #   # │ mean       ┆ 3.0      │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┤
     #   # │ std        ┆ 1.581139 │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┤
     #   # │ count      ┆ 5.0      │
     #   # └────────────┴──────────┘
     #
@@ -455,9 +461,7 @@ module Polars
     #   # │ str        ┆ i64   │
     #   # ╞════════════╪═══════╡
     #   # │ unique     ┆ 4     │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌┤
     #   # │ null_count ┆ 1     │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌┤
     #   # │ count      ┆ 5     │
     #   # └────────────┴───────┘
     def describe
@@ -660,13 +664,11 @@ module Polars
     #   # │ u8  ┆ u8  ┆ u8  │
     #   # ╞═════╪═════╪═════╡
     #   # │ 1   ┆ 0   ┆ 0   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ 0   ┆ 1   ┆ 0   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ 0   ┆ 0   ┆ 1   │
     #   # └─────┴─────┴─────┘
-    def to_dummies
-      Utils.wrap_df(_s.to_dummies)
+    def to_dummies(separator: "_")
+      Utils.wrap_df(_s.to_dummies(separator))
     end
 
     # Count the unique values in a Series.
@@ -687,9 +689,7 @@ module Polars
     #   # │ i64 ┆ u32    │
     #   # ╞═════╪════════╡
     #   # │ 1   ┆ 1      │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌╌╌┤
     #   # │ 2   ┆ 2      │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌╌╌╌┤
     #   # │ 3   ┆ 1      │
     #   # └─────┴────────┘
     def value_counts(sort: false)
@@ -1285,8 +1285,12 @@ module Polars
     #   Expression or scalar value.
     #
     # @return [Integer]
-    def search_sorted(element)
-      Polars.select(Polars.lit(self).search_sorted(element))[0, 0]
+    def search_sorted(element, side: "any")
+      if element.is_a?(Integer) || element.is_a?(Float)
+        return Polars.select(Polars.lit(self).search_sorted(element, side: side)).item
+      end
+      element = Series.new(element)
+      Polars.select(Polars.lit(self).search_sorted(element, side: side)).to_series
     end
 
     # Get unique elements in series.
@@ -1500,7 +1504,7 @@ module Polars
     #   sets = Polars::Series.new("sets", [[1, 2, 3], [1, 2], [9, 10]])
     #   # =>
     #   # shape: (3,)
-    #   # Series: 'sets' [list]
+    #   # Series: 'sets' [list[i64]]
     #   # [
     #   #         [1, 2, 3]
     #   #         [1, 2]