groupby().agg() silently ignores weights, producing incorrect results
Los mantenedores suelen responder en 1 día
Nadie ha tomado este issue todavía.
Evaluación
- Dificultad
- 3/5
- Tiempo estimado
- 1-2 días
- Aptitud para principiantes
- 65/100
- Tipo de issue
- Error
- Claridad
- Bien especificado
- Estado de actividad
- Activo
Línea de trabajo
The issue is in microdf/microdataframe.py lines 644-680, where the MicroDataFrameGroupBy class does not override .agg(). Start by reading the existing weighted method overrides (like sum). Determine how to parse aggregation specs and apply weights, or implement a clear error/warning. Run the provided test case to verify the fix.
Escrito por el modelo de indexación a partir del texto del issue.
Descripción
Problem
MicroDataFrame's groupby operations silently ignore weights when using .agg(), producing incorrect (unweighted) results without any warning or error.
Example
import microdf as mdf
import numpy as np
df = mdf.MicroDataFrame(
{"group": ["A", "A", "B", "B"], "value": [10, 20, 30, 40]},
weights=np.array([2, 3, 1, 4])
)
# CORRECT (weighted):
df.groupby("group").value.sum()
# A: 10*2 + 20*3 = 80.0
# B: 30*1 + 40*4 = 190.0
# INCORRECT (unweighted) - no warning!
df.groupby("group").agg({'value': 'sum'})
# A: 10 + 20 = 30 ❌
# B: 30 + 40 = 70 ❌
Impact
This is a critical data correctness issue because:
- Silent failure: No error or warning - just wrong numbers
- Natural usage pattern:
.agg()is a standard pandas idiom for multi-column aggregation - Plausible results: The unweighted numbers look reasonable, making bugs hard to detect
- Real-world consequences: Users analyzing survey data (CPS, ACS, etc.) will get wildly incorrect population estimates
Root Cause
Looking at microdf/microdataframe.py:644-680, the MicroDataFrameGroupBy class:
- ✓ Overrides specific methods like
sum(),mean(), etc. to apply weights - ✗ Does NOT override
.agg()or.aggregate(), so they fall back to pandas' unweighted implementation
Proposed Solutions
Option 1: Override .agg() to apply weights (Best)
- Implement
MicroDataFrameGroupBy.agg()to properly handle weights - Parse the aggregation specifications and route to weighted methods
Option 2: Raise an error (Safer than current behavior)
def agg(self, *args, **kwargs):
raise NotImplementedError(
"MicroDataFrameGroupBy.agg() does not support weights. "
"Use df.groupby(col).column.sum() instead."
)
Option 3: Emit a loud warning
def agg(self, *args, **kwargs):
warnings.warn(
"MicroDataFrameGroupBy.agg() ignores weights! Results will be unweighted.",
UserWarning,
stacklevel=2
)
return super().agg(*args, **kwargs)
Related Issues
This extends #193, which identified similar problems with .groupby()[[cols]].sum() but didn't specifically address .agg().
Additional Test Cases Needed
def test_agg_with_weights():
"""Test that .agg() applies weights correctly or raises an error"""
df = mdf.MicroDataFrame(
{"group": ["A", "A", "B"], "value": [10, 20, 30]},
weights=np.array([2, 3, 4])
)
# These should either work correctly or raise NotImplementedError
result = df.groupby("group").agg({'value': 'sum'})
# If implemented, should equal weighted sums
# A: 10*2 + 20*3 = 80
# B: 30*4 = 120
expected = pd.DataFrame({'value': [80.0, 120.0]}, index=['A', 'B'])
expected.index.name = 'group'
# Should NOT be unweighted sums (30, 30)
assert not result.equals(pd.DataFrame({'value': [30, 30]}))
Priority
HIGH - This is a data correctness bug that produces silently wrong results in a library designed for weighted survey analysis.
- Lenguaje dominante
- Python
- Estrellas
- 16
- Forks
- 10
- Merge medio
- 5 d 3 h
- PR fusionados (30 d)
- 21
Preparar el entorno
Primeros pasos
- Lee el issue completo y luego la guía de contribución del proyecto.
- Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
- Haz un fork del repositorio y trabaja en una rama.
- Abre un pull request que haga referencia al número del issue.
Más de PolicyEngine/microdf
-
docs/examples.md still says MicroDataFrame.cov() and .corr() are unweightedPosiblemente ocupada @juaristi22 la tomó hace 4 días. Abierto
PolicyEngine/microdf#335 · 1 asignado ·
Los mantenedores suelen responder en 1 día
-
Poverty gap docstrings overclaim FGT indices, and the poverty estimators have no testsPosiblemente ocupada @juaristi22 la tomó hace 4 días. Abierto
PolicyEngine/microdf#334 · 1 asignado ·
Los mantenedores suelen responder en 1 día
-
Fail closed: aggregation and construction paths that silently return unweighted resultsPosiblemente ocupada @juaristi22 la tomó hace 4 días. Abiertobug
PolicyEngine/microdf#333 · 1 asignado ·
Los mantenedores suelen responder en 1 día
-
Dificultad 5/5 Más de una semana Aptitud para principiantes 30/100
PolicyEngine/microdf#314 ·
Los mantenedores suelen responder en 1 día
-
Dificultad 2/5 1-3 horas Aptitud para principiantes 45/100
PolicyEngine/microdf#223 ·
Los mantenedores suelen responder en 1 día
Todos los issues de PolicyEngine/microdf
Issues similares
-
needs triage
Dificultad 2/5 1-3 horas Aptitud para principiantes 78/100
Los mantenedores suelen responder en 2 días
-
Dificultad 2/5 1-3 horas Aptitud para principiantes 82/100
openvinotoolkit/openvino_notebooks#3665 ·
Los mantenedores suelen responder en 1 día
-
bug
Dificultad 2/5 1-3 horas Aptitud para principiantes 86/100
Los mantenedores suelen responder en 1 día
-
docs
Dificultad 2/5 1-3 horas Aptitud para principiantes 88/100
Los mantenedores suelen responder en 1 día
-
benchmark-gap
Dificultad 2/5 1-3 horas Aptitud para principiantes 78/100
Los mantenedores suelen responder en 1 día