Merge pull request #351 from lincc-frameworks/fix-350

Fix reduce for non-uniq idx

Author: hombit

src/nested_pandas/nestedframe/core.py

@@ -1870,7 +1870,8 @@ def reduce(self, func, *args, infer_nesting=True, append_columns=False, **kwargs
                 nested_col = pack_lists(rename_df, name=layer)
                 results_nf = results_nf[
                     [col for col in results_nf.columns if not col.startswith(f"{layer}.")]
-                ].join(nested_col)
+                ]
+                results_nf[layer] = nested_col
 
         if append_columns:
             # Append the results to the original NestedFrame

tests/nested_pandas/nestedframe/test_nestedframe.py

@@ -2199,3 +2199,11 @@ def test_issue294():
     # Failed with a ValueError in the original issue
     nf["nested.c"] = nf["c"]
     nf["nested.mag"] = -2.5 * np.log10(nf["nested.flux"])
+
+
+def test_issue350():
+    """https://github.com/lincc-frameworks/nested-pandas/issues/350"""
+    nf = generate_data(3, 2)
+    nf = nf.set_index(np.array([100, 100, 101]))
+    result = nf.reduce(lambda flux: {"new.flux": flux}, "nested.flux")
+    assert len(result) == 3