Simulação de mudança de temperatura para partidas Cosmos

simulacao.py

from random import sample
from collections import defaultdict


RODADAS = 7
EVENTOS = [-2, -1, -1, -1, -1, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2]


def simular_partida():
  eventos_da_partida = sample(EVENTOS, RODADAS)
  delta_temperatura = sum(eventos_da_partida)
  return delta_temperatura


def calcular_histograma(partidas):
  histograma = defaultdict(lambda: 0)
  for _ in range(partidas):
    delta_temperatura = simular_partida()
    histograma[delta_temperatura] = histograma[delta_temperatura] + 1
  return histograma


histograma = calcular_histograma(100000)
print(histograma)