microdf-python 1.0.0__tar.gz → 1.0.1__tar.gz

{microdf_python-1.0.0/microdf_python.egg-info → microdf_python-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microdf-python
-Version: 1.0.0
+Version: 1.0.1
 Summary: Weighted pandas DataFrames and Series for survey microdata
 Author-email: Max Ghenis <max@ubicenter.org>
 License: MIT

{microdf_python-1.0.0 → microdf_python-1.0.1}/microdf/microdataframe.py

@@ -218,13 +218,15 @@ class MicroDataFrame(pd.DataFrame):
     def __getitem__(
         self, key: Union[str, List]
     ) -> Union[pd.Series, pd.DataFrame]:
+        # Let pandas handle the initial slicing
         result = super().__getitem__(key)
+
+        # If the result is a DataFrame, re-synchronize the weights
         if isinstance(result, pd.DataFrame):
-            try:
-                weights = self.weights[key]
-            except Exception:
-                weights = self.weights
-            return MicroDataFrame(result, weights=weights)
+            new_weights = self.weights.reindex(result.index)
+            return MicroDataFrame(result, weights=new_weights)
+
+        # Otherwise, the result is a Series or a scalar, so just return it
         return result
 
     def catch_series_relapse(self) -> None:

{microdf_python-1.0.0 → microdf_python-1.0.1}/microdf/tests/test_microseries_dataframe.py

@@ -24,6 +24,15 @@ def test_df_init() -> None:
     assert df.a.mean() == np.average(arr, weights=w)
 
 
+def test_handles_empty_index() -> None:
+    arr = np.array([0, 1, 1])
+    w = np.array([3, 0, 9])
+    df = mdf.MicroDataFrame({"a": arr}, weights=w)
+
+    empty_index = pd.Index([])
+    df[empty_index]  # Implicit assert; checking for ValueError
+
+
 def test_series_getitem() -> None:
     arr = np.array([0, 1, 1])
     w = np.array([3, 0, 9])

{microdf_python-1.0.0 → microdf_python-1.0.1/microdf_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microdf-python
-Version: 1.0.0
+Version: 1.0.1
 Summary: Weighted pandas DataFrames and Series for survey microdata
 Author-email: Max Ghenis <max@ubicenter.org>
 License: MIT

{microdf_python-1.0.0 → microdf_python-1.0.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "microdf-python"
-version = "1.0.0"
+version = "1.0.1"
 description = "Weighted pandas DataFrames and Series for survey microdata"
 readme = "README.md"
 authors = [