f49809322491c857d435bb12f5de7b09aac2d181,linearmodels/panel/data.py,PanelData,mean,#PanelData#Any#Any#,378

Before Change

            mu = np.nanmean(v, axis=axis)
        else:
            w = weights.values3d
            w = w * np.isfinite(v)
            wv = w * v
            mu = np.nansum(wv, axis=axis)
            mu /= np.nansum(w, axis=axis)

After Change

            w = weights.values2d
            frame = self._frame.copy()
            frame = w * frame
            weighted_sum = frame.groupby(level=level).sum()
            frame.iloc[:, :] = w
            sum_weights = frame.groupby(level=level).sum()
            mu = weighted_sum / sum_weights

        reindex = self.entities if group == "entity" else self.time
        out = mu.loc[reindex]

Instances