commit 9dcc442584b752f77291e0f6dc3c089fd23708de
parent 0405e3815c0fbcf4162b8f25aa978fa6c6d8a982
Author: NunoSempere <nuno.semperelh@protonmail.com>
Date: Sun, 14 Apr 2024 15:33:36 -0400
add results histogram
Diffstat:
| M | main.go | | | 38 | ++++++++++++++++++++++++++++++++++++-- |
1 file changed, 36 insertions(+), 2 deletions(-)
diff --git a/main.go b/main.go
@@ -373,6 +373,7 @@ func sampleFromState(state State) VotesForEachParty {
}
}
+/* Simulate election */
func simulateElection(states []State) int {
republican_seats := 0
@@ -383,6 +384,36 @@ func simulateElection(states []State) int {
return republican_seats
}
+/* Histogram */
+func barString(n int) string {
+ str := ""
+ for i := 0; i < n; i++ {
+ str += "█"
+ }
+ return str
+}
+
+func printElectoralCollegeHistogram(samples []int) {
+
+ histogram := [538]int{}
+ for _, sample := range samples {
+ histogram[sample]++
+ }
+
+ max_count := 0
+ for _, count := range histogram {
+ if count > max_count {
+ max_count = count
+ }
+ }
+
+ for i, count := range histogram {
+ bar_length := (count * 100) / max_count // Assuming max_count bar length is 50 characters
+ fmt.Printf("[ %2d, %4d): %s %d\n", i, i+1, barString(bar_length), count)
+ }
+
+}
+
func main() {
states, err := readStates()
if err != nil {
@@ -390,19 +421,22 @@ func main() {
return
}
- n_sims := 10_000
+ n_sims := 100_000
printStates(states)
p_republicans := 0.0
+ results := make([]int, n_sims)
for i := 0; i < n_sims; i++ {
republican_seats := simulateElection(states)
+ results[i] = republican_seats
if republican_seats >= 270 {
p_republicans++
}
- fmt.Printf("%d - %t\n", republican_seats, republican_seats >= 270)
+ // fmt.Printf("%d - %t\n", republican_seats, republican_seats >= 270)
}
p_republicans = p_republicans / float64(n_sims)
fmt.Printf("\n\n\n%% republicans: %f\n", p_republicans)
+ printElectoralCollegeHistogram(results)
}
