kumisbespassung/lotto/lotto.py

62 lines
2 KiB
Python
Raw Normal View History

2017-11-12 16:44:32 +00:00
#!/usr/bin/env python3
import random
import time
import argparse
import matplotlib.pyplot
def draw_sample(limit = 45, count = 7, system = False):
rand = random.SystemRandom() if system else random
return rand.sample(range(1, limit + 1), count)
def generate_stats(results):
numbers = {}
for result in results:
for number in result:
try:
numbers[number] += 1
except KeyError:
numbers[number] = 1
return numbers
def run_draws(draws = 1, benchmark = False, statistics = False, *args, **kwargs):
results = []
start = time.time()
for __ in range(0, draws):
results += [draw_sample(*args, **kwargs)]
end = time.time()
if statistics:
results = generate_stats(results)
return (end - start) if benchmark else results
def generate_plot(results, outfile):
matplotlib.pyplot.bar(list(results.keys()), results.values())
matplotlib.pyplot.savefig(outfile)
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--count", "-c", type=int, help="number of numbers per draw", default=7)
parser.add_argument("--draws", "-d", type=int, help="number of draws to simulate", default=1)
parser.add_argument("--limit", "-l", type=int, help="highest number to be drawn (lowest is always 1)", default=45)
parser.add_argument("--system", "-s", action="store_true", help="use more secure source of randomness (considerably slower)")
formats = parser.add_mutually_exclusive_group()
formats.add_argument("--benchmark", "-b", action="store_true", help="output time needed to process draws instead of numbers drawn")
formats.add_argument("--analysis", "-a", action="store_true", help="output statistics about draws instead of numbers drawn")
formats.add_argument("--graph", "-g", type=str, help="generate chart in file with statistics about draws instead of numbers drawn", metavar="FILE")
args = parser.parse_args()
output = run_draws(args.draws, args.benchmark, args.analysis or not not args.graph, args.limit, args.count, args.system)
if args.graph:
generate_plot(output, args.graph)
else:
print(output)