Python 计算纳什-萨特克利夫效率

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

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

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