Python 分组加权平均处理NAN

我正在计算一些字段的加权平均数，在这里的建议帮助下，我写了以下内容：

wm = lambda x: np.average(x, weights=dfstrat.loc[x.index, 'pop'])
dfdom = dfstrat.groupby('dom').agg(
    dom_n = ('n', 'sum'),
    dom_meanAGkWh = ('meanAGkwh', wm),
    dom_meanAGkW = ('meanAGkw', wm))

---

这非常有效，除非“meanAGkwh”或“meanAGkw”在其中一行中有一个nan。然后，该“dom”的返回结果是一个nan。任何关于如何忽略NAN的建议。对于特定的dom，meanAGkwh可能有一行数据，其中meanAGkwh为nan。

尝试使用numpy掩码平均值：