fix(rust, python): Add list inner dtype when printing Series (#6233)

pola-rs · Jan 15, 2023 · a445cfc · a445cfc
1 parent 6a1d8bf
commit a445cfc
Show file tree

Hide file tree

Showing 7 changed files with 17 additions and 16 deletions.
diff --git a/polars/polars-core/src/fmt.rs b/polars/polars-core/src/fmt.rs
@@ -259,7 +259,8 @@ impl Debug for Series {
                 format_array!(f, self.duration().unwrap(), &dt, self.name(), "Series")
             }
             DataType::List(_) => {
-                format_array!(f, self.list().unwrap(), "list", self.name(), "Series")
+                let dt = format!("{}", self.dtype());
+                format_array!(f, self.list().unwrap(), &dt, self.name(), "Series")
             }
             #[cfg(feature = "object")]
             DataType::Object(_) => format_object_array(f, self, self.name(), "Series"),
@@ -940,7 +941,7 @@ mod test {
 
         assert_eq!(
             r#"shape: (2,)
-Series: 'a' [list]
+Series: 'a' [list[i32]]
 [
 	[1, 2, 3]
 	null

diff --git a/py-polars/polars/internals/expr/list.py b/py-polars/polars/internals/expr/list.py
@@ -489,7 +489,7 @@ def diff(self, n: int = 1, null_behavior: NullBehavior = "ignore") -> pli.Expr:
         >>> s = pl.Series("a", [[1, 2, 3, 4], [10, 2, 1]])
         >>> s.arr.diff()
         shape: (2,)
-        Series: 'a' [list]
+        Series: 'a' [list[i64]]
         [
             [null, 1, ... 1]
             [null, -8, -1]
@@ -512,7 +512,7 @@ def shift(self, periods: int = 1) -> pli.Expr:
         >>> s = pl.Series("a", [[1, 2, 3, 4], [10, 2, 1]])
         >>> s.arr.shift()
         shape: (2,)
-        Series: 'a' [list]
+        Series: 'a' [list[i64]]
         [
             [null, 1, ... 3]
             [null, 10, 2]
@@ -540,7 +540,7 @@ def slice(
         >>> s = pl.Series("a", [[1, 2, 3, 4], [10, 2, 1]])
         >>> s.arr.slice(1, 2)
         shape: (2,)
-        Series: 'a' [list]
+        Series: 'a' [list[i64]]
         [
             [2, 3]
             [2, 1]
@@ -565,7 +565,7 @@ def head(self, n: int | str | pli.Expr = 5) -> pli.Expr:
         >>> s = pl.Series("a", [[1, 2, 3, 4], [10, 2, 1]])
         >>> s.arr.head(2)
         shape: (2,)
-        Series: 'a' [list]
+        Series: 'a' [list[i64]]
         [
             [1, 2]
             [10, 2]
@@ -588,7 +588,7 @@ def tail(self, n: int | str | pli.Expr = 5) -> pli.Expr:
         >>> s = pl.Series("a", [[1, 2, 3, 4], [10, 2, 1]])
         >>> s.arr.tail(2)
         shape: (2,)
-        Series: 'a' [list]
+        Series: 'a' [list[i64]]
         [
             [3, 4]
             [2, 1]

diff --git a/py-polars/polars/internals/series/list.py b/py-polars/polars/internals/series/list.py
@@ -193,7 +193,7 @@ def diff(self, n: int = 1, null_behavior: NullBehavior = "ignore") -> pli.Series
         >>> s = pl.Series("a", [[1, 2, 3, 4], [10, 2, 1]])
         >>> s.arr.diff()
         shape: (2,)
-        Series: 'a' [list]
+        Series: 'a' [list[i64]]
         [
             [null, 1, ... 1]
             [null, -8, -1]
@@ -215,7 +215,7 @@ def shift(self, periods: int = 1) -> pli.Series:
         >>> s = pl.Series("a", [[1, 2, 3, 4], [10, 2, 1]])
         >>> s.arr.shift()
         shape: (2,)
-        Series: 'a' [list]
+        Series: 'a' [list[i64]]
         [
             [null, 1, ... 3]
             [null, 10, 2]
@@ -240,7 +240,7 @@ def slice(self, offset: int, length: int | None = None) -> pli.Series:
         >>> s = pl.Series("a", [[1, 2, 3, 4], [10, 2, 1]])
         >>> s.arr.slice(1, 2)
         shape: (2,)
-        Series: 'a' [list]
+        Series: 'a' [list[i64]]
         [
             [2, 3]
             [2, 1]
@@ -262,7 +262,7 @@ def head(self, n: int = 5) -> pli.Series:
         >>> s = pl.Series("a", [[1, 2, 3, 4], [10, 2, 1]])
         >>> s.arr.head(2)
         shape: (2,)
-        Series: 'a' [list]
+        Series: 'a' [list[i64]]
         [
             [1, 2]
             [10, 2]
@@ -284,7 +284,7 @@ def tail(self, n: int = 5) -> pli.Series:
         >>> s = pl.Series("a", [[1, 2, 3, 4], [10, 2, 1]])
         >>> s.arr.tail(2)
         shape: (2,)
-        Series: 'a' [list]
+        Series: 'a' [list[i64]]
         [
             [3, 4]
             [2, 1]

diff --git a/py-polars/polars/internals/series/series.py b/py-polars/polars/internals/series/series.py
@@ -2351,7 +2351,7 @@ def is_in(self, other: Series | Sequence[Any]) -> Series:
         >>> optional_members = pl.Series("optional_members", [1, 2, 3])
         >>> print(sets)
         shape: (3,)
-        Series: 'sets' [list]
+        Series: 'sets' [list[i64]]
         [
             [1, 2, 3]
             [1, 2]

diff --git a/py-polars/polars/internals/series/string.py b/py-polars/polars/internals/series/string.py
@@ -406,7 +406,7 @@ def extract_all(self, pattern: str | pli.Series) -> pli.Series:
         >>> s = pl.Series("foo", ["123 bla 45 asd", "xyz 678 910t"])
         >>> s.str.extract_all(r"(\d+)")
         shape: (2,)
-        Series: 'foo' [list]
+        Series: 'foo' [list[str]]
         [
             ["123", "45"]
             ["678", "910"]

diff --git a/py-polars/tests/unit/io/test_other.py b/py-polars/tests/unit/io/test_other.py
@@ -39,7 +39,7 @@ def test_date_list_fmt() -> None:
     assert (
         str(df.groupby("index", maintain_order=True).agg(pl.col("mydate"))["mydate"])
         == """shape: (3,)
-Series: 'mydate' [list]
+Series: 'mydate' [list[date]]
 [
 	[2020-01-01]
 	[2020-01-02]

diff --git a/py-polars/tests/unit/test_df.py b/py-polars/tests/unit/test_df.py
@@ -1664,7 +1664,7 @@ def test_groupby_cat_list() -> None:
     assert (
         str(grouped)
         == """shape: (3,)
-Series: 'cat_column' [list]
+Series: 'cat_column' [list[cat]]
 [
 	["a", "b"]
 	["b", "a"]