如何有效地计算给定数计数器的方差或标准差?

Question

例如给出下一个直方图和箱子:

import numpy as np
hist = np.array([1,1,2,1,2])
bins = np.array([0,1,2,3,4 ])

¿什么是最有效的方法来计算方差?一种方法是重新创建数组并将其传递给np.var函数:

import numpy as np
np.var(np.array([0, 1, 2, 2, 3, 4, 4]))

然而，我认为这不是很有效。

Answer 1

所以你可以重写这个公式:

counts  = hist.sum()
mean = (hist*bins).sum() / counts
sum_squares = (bins**2 * hist).sum()
var = sum_squares/counts - mean ** 2
# test
np.isclose(var, np.var(np.repeat(bins, hist)))

输出True.