GitHub

Original file line numberDiff line numberDiff line change

@@ -172,13 +172,13 @@ We verify the above and compute the mean and variance using `numpy`.

172172
173173

```{code-cell} ipython3

174174

Benford_pmf = np.array([np.log10(1+1/d) for d in range(1,10)])

175-

k = np.array(range(1,10))

175+

k = np.arange(1, 10)

176176
177177

# mean

178-

mean = np.sum(Benford_pmf * k)

178+

mean = k @ Benford_pmf

179179
180180

# variance

181-

var = np.sum([(k-mean)**2 * Benford_pmf])

181+

var = ((k - mean) ** 2) @ Benford_pmf

182182
183183

# verify sum to 1

184184

print(np.sum(Benford_pmf))

Read the original on github.com ↗