From 97ea0a140ac1377c7d0ba05fddc2633665f95579 Mon Sep 17 00:00:00 2001 From: suda-yuga Date: Wed, 25 Jun 2025 21:22:50 +0900 Subject: [PATCH 1/2] Replace np.sum(a * b) with a @ b --- lectures/prob_matrix.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lectures/prob_matrix.md b/lectures/prob_matrix.md index 783a9a262..6749cc43c 100644 --- a/lectures/prob_matrix.md +++ b/lectures/prob_matrix.md @@ -756,13 +756,13 @@ We verify the above and compute the mean and variance using `numpy`. ```{code-cell} ipython3 Benford_pmf = np.array([np.log10(1+1/d) for d in range(1,10)]) -k = np.array(range(1,10)) +k = np.arange(1, 10) # mean -mean = np.sum(Benford_pmf * k) +mean = Benford_pmf @ k # variance -var = np.sum([(k-mean)**2 * Benford_pmf]) +var = ((k - mean) ** 2) @ Benford_pmf # verify sum to 1 print(np.sum(Benford_pmf)) From a381627984155ecfc7ee8474c849443d048c1c12 Mon Sep 17 00:00:00 2001 From: suda-yuga Date: Mon, 30 Jun 2025 16:29:05 +0900 Subject: [PATCH 2/2] Update lectures/prob_matrix.md Co-authored-by: Daisuke Oyama --- lectures/prob_matrix.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lectures/prob_matrix.md b/lectures/prob_matrix.md index 6749cc43c..0444d3a8b 100644 --- a/lectures/prob_matrix.md +++ b/lectures/prob_matrix.md @@ -759,7 +759,7 @@ Benford_pmf = np.array([np.log10(1+1/d) for d in range(1,10)]) k = np.arange(1, 10) # mean -mean = Benford_pmf @ k +mean = k @ Benford_pmf # variance var = ((k - mean) ** 2) @ Benford_pmf