fix(python): allow from pandas null structs (#6430)

pola-rs · Jan 25, 2023 · 7e5ac26 · 7e5ac26
1 parent 3adef12
commit 7e5ac26
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/polars/polars-core/src/series/from.rs b/polars/polars-core/src/series/from.rs
@@ -338,9 +338,14 @@ impl Series {
                     let new_values = struct_arr
                         .values()
                         .iter()
-                        .map(|arr| match arr.validity() {
-                            None => arr.with_validity(Some(validity.clone())),
-                            Some(arr_validity) => arr.with_validity(Some(arr_validity & validity)),
+                        .map(|arr| match arr.data_type() {
+                            ArrowDataType::Null => arr.clone(),
+                            _ => match arr.validity() {
+                                None => arr.with_validity(Some(validity.clone())),
+                                Some(arr_validity) => {
+                                    arr.with_validity(Some(arr_validity & validity))
+                                }
+                            },
                         })
                         .collect();
 

diff --git a/py-polars/tests/unit/test_interop.py b/py-polars/tests/unit/test_interop.py
@@ -584,3 +584,16 @@ def test_arrow_list_null_5697() -> None:
     assert pl.from_arrow(pa_table,).schema == {  # type: ignore[union-attr]
         "mycol": pl.List(pl.Null)
     }
+
+
+def test_from_pandas_null_struct_6412() -> None:
+    data = [
+        {
+            "a": {
+                "b": None,
+            },
+        },
+        {"a": None},
+    ]
+    df_pandas = pd.DataFrame(data)
+    assert pl.from_pandas(df_pandas).to_dict(False) == {"a": [{"b": None}, {"b": None}]}