计算纳什效率

Question

我有两个数组，它们的形状都是（220，6），如何计算NSE（纳什-萨特克利夫效率）？ 我知道当它有一列时如何计算，如下所示：

denominator = np.sum((a1 - np.mean(a1)) ** 2)
numerator = np.sum((a2 - a1) ** 2)
nse_val = 1 - numerator / denominator

这是否也适用于具有更多列的数组

Answer 1

这是我用于Nash-Sutcliffe效率（NSE）计算的函数

def nse(predictions, targets):
    return (1-(np.sum((predictions-targets)**2)/np.sum((targets-np.mean(targets))**2))

针对1d和2d numpy阵列进行了测试，没有任何NaN，可以正常工作