refactor(rust, python)!: iso weekday (#5598)

pola-rs · Nov 23, 2022 · 2123266 · 2123266
1 parent ec79597
commit 2123266
Show file tree

Hide file tree

Showing 6 changed files with 19 additions and 19 deletions.
diff --git a/polars/polars-time/src/chunkedarray/datetime.rs b/polars/polars-time/src/chunkedarray/datetime.rs
@@ -66,8 +66,8 @@ pub trait DatetimeMethods: AsDatetime {
         cast_and_apply(self.as_datetime(), temporal::month)
     }
 
-    /// Extract weekday from underlying NaiveDateTime representation.
-    /// Returns the weekday number where monday = 0 and sunday = 6
+    /// Extract ISO weekday from underlying NaiveDateTime representation.
+    /// Returns the weekday number where monday = 1 and sunday = 7
     fn weekday(&self) -> UInt32Chunked {
         let ca = self.as_datetime();
         let f = match ca.time_unit() {

diff --git a/polars/polars-time/src/chunkedarray/kernels.rs b/polars/polars-time/src/chunkedarray/kernels.rs
@@ -20,7 +20,7 @@ trait PolarsIso {
 
 impl PolarsIso for NaiveDateTime {
     fn p_weekday(&self) -> u32 {
-        self.weekday() as u32
+        self.weekday() as u32 + 1
     }
     fn week(&self) -> u32 {
         self.iso_week().week()
@@ -32,7 +32,7 @@ impl PolarsIso for NaiveDateTime {
 
 impl PolarsIso for NaiveDate {
     fn p_weekday(&self) -> u32 {
-        self.weekday() as u32
+        self.weekday() as u32 + 1
     }
     fn week(&self) -> u32 {
         self.iso_week().week()

diff --git a/py-polars/polars/internals/expr/datetime.py b/py-polars/polars/internals/expr/datetime.py
@@ -517,7 +517,7 @@ def weekday(self) -> pli.Expr:
 
         Applies to Date and Datetime columns.
 
-        Returns the weekday number where monday = 0 and sunday = 6
+        Returns the ISO weekday number where monday = 1 and sunday = 7
 
         Returns
         -------
@@ -555,11 +555,11 @@ def weekday(self) -> pli.Expr:
         │ ---     ┆ ---          ┆ ---         │
         │ u32     ┆ u32          ┆ u32         │
         ╞═════════╪══════════════╪═════════════╡
-        │ 0       ┆ 1            ┆ 1           │
+        │ 1       ┆ 1            ┆ 1           │
         ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-        │ 3       ┆ 4            ┆ 4           │
+        │ 4       ┆ 4            ┆ 4           │
         ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-        │ 6       ┆ 7            ┆ 7           │
+        │ 7       ┆ 7            ┆ 7           │
         └─────────┴──────────────┴─────────────┘
 
         """
@@ -610,11 +610,11 @@ def day(self) -> pli.Expr:
         │ ---     ┆ ---          ┆ ---         │
         │ u32     ┆ u32          ┆ u32         │
         ╞═════════╪══════════════╪═════════════╡
-        │ 0       ┆ 1            ┆ 1           │
+        │ 1       ┆ 1            ┆ 1           │
         ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-        │ 3       ┆ 4            ┆ 4           │
+        │ 4       ┆ 4            ┆ 4           │
         ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-        │ 6       ┆ 7            ┆ 7           │
+        │ 7       ┆ 7            ┆ 7           │
         └─────────┴──────────────┴─────────────┘
 
         """
@@ -665,11 +665,11 @@ def ordinal_day(self) -> pli.Expr:
         │ ---     ┆ ---          ┆ ---         │
         │ u32     ┆ u32          ┆ u32         │
         ╞═════════╪══════════════╪═════════════╡
-        │ 0       ┆ 1            ┆ 1           │
+        │ 1       ┆ 1            ┆ 1           │
         ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-        │ 3       ┆ 4            ┆ 4           │
+        │ 4       ┆ 4            ┆ 4           │
         ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-        │ 6       ┆ 7            ┆ 7           │
+        │ 7       ┆ 7            ┆ 7           │
         └─────────┴──────────────┴─────────────┘
 
         """

diff --git a/py-polars/polars/internals/series/datetime.py b/py-polars/polars/internals/series/datetime.py
@@ -342,7 +342,7 @@ def weekday(self) -> pli.Series:
 
         Applies to Date and Datetime columns.
 
-        Returns the weekday number where monday = 0 and sunday = 6
+        Returns the ISO weekday number where monday = 1 and sunday = 7
 
         Returns
         -------
@@ -370,13 +370,13 @@ def weekday(self) -> pli.Series:
         shape: (7,)
         Series: '' [u32]
         [
-                0
                 1
                 2
                 3
                 4
                 5
                 6
+                7
         ]
 
         """

diff --git a/py-polars/tests/unit/test_datelike.py b/py-polars/tests/unit/test_datelike.py
@@ -1606,9 +1606,9 @@ def test_weekday() -> None:
 
     time_units: list[TimeUnit] = ["ns", "us", "ms"]
     for tu in time_units:
-        assert s.dt.cast_time_unit(tu).dt.weekday()[0] == 0
+        assert s.dt.cast_time_unit(tu).dt.weekday()[0] == 1
 
-    assert s.cast(pl.Date).dt.weekday()[0] == 0
+    assert s.cast(pl.Date).dt.weekday()[0] == 1
 
 
 @pytest.mark.skip(reason="from_dicts cannot yet infer timezones")

diff --git a/py-polars/tests/unit/test_series.py b/py-polars/tests/unit/test_series.py
@@ -1829,7 +1829,7 @@ def test_dt_year_month_week_day_ordinal_day() -> None:
     verify_series_and_expr_api(a, exp, "dt.year")
 
     verify_series_and_expr_api(a, pl.Series("a", [5, 10, 2], dtype=UInt32), "dt.month")
-    verify_series_and_expr_api(a, pl.Series("a", [0, 4, 1], dtype=UInt32), "dt.weekday")
+    verify_series_and_expr_api(a, pl.Series("a", [1, 5, 2], dtype=UInt32), "dt.weekday")
     verify_series_and_expr_api(a, pl.Series("a", [21, 40, 8], dtype=UInt32), "dt.week")
     verify_series_and_expr_api(a, pl.Series("a", [19, 4, 20], dtype=UInt32), "dt.day")
     verify_series_and_expr_api(