results_stats.py

#!/usr/bin/env python3

import sys
import statistics
import numpy


def main():
    import matplotlib.pyplot

    games = list(collect_data(sorted(sys.argv[1:])))
    seeds = [v['seed'] for v in games]
    players = dict(
        first=dict(places=[], scores=[], score_diffs=[]),
        second=dict(places=[], scores=[], score_diffs=[]),
    )
    for r in games:
        for k, p in players.items():
            p['places'].append(r[k]['place'])
            p['scores'].append(r[k]['score'])
        players['first']['score_diffs'].append(r['first']['score'] - r['second']['score'])
        players['second']['score_diffs'].append(r['second']['score'] - r['first']['score'])
    stats = dict(
        _1=[sum(w == 1 for w in v['places']) for v in players.values()],
        _2=[sum(w == 2 for w in v['places']) for v in players.values()],
        total_score=[sum(v['scores']) for v in players.values()],
        min_score=[min(v['scores']) for v in players.values()],
        max_score=[max(v['scores']) for v in players.values()],
        mean_score=[statistics.mean(v['scores']) for v in players.values()],
        median_score=[statistics.median(v['scores']) for v in players.values()],
        stdev_score=[statistics.stdev(v['scores']) for v in players.values()],
        q95_score=[numpy.quantile(v['scores'], 0.95) for v in players.values()],
        min_score_diff=[min(v['score_diffs']) for v in players.values()],
        max_score_diff=[max(v['score_diffs']) for v in players.values()],
        mean_score_diff=[statistics.mean(v['score_diffs']) for v in players.values()],
        median_score_diff=[statistics.median(v['score_diffs']) for v in players.values()],
        stdev_score_diff=[statistics.stdev(v['score_diffs']) for v in players.values()],
    )
    row('games', len(games))
    row('unique seeds', len(set(seeds)))
    print()
    row('', *(list(players.keys()) + ['ratio (second/first)']))
    for k, v in stats.items():
        row(k, *(v + [ratio(v)]))
    print()
    print(*seeds)
    fig, ax = matplotlib.pyplot.subplots()
    ax.set_title('scores distribution')
    bins = numpy.arange(0, max(max(v['scores']) for v in players.values()) + 1)
    for k, v in players.items():
        ax.hist(v['scores'], bins=bins, label=k, alpha=0.5)
        ax.set_xticks(bins)
        ax.grid(True)
        ax.legend()
    _1 = {k: [0] for k in players.keys()}
    fig, ax = matplotlib.pyplot.subplots()
    ax.set_title('place 1 dynamic')
    for g in games:
        for k, v in _1.items():
            v.append(v[-1] + (g[k]['place'] == 1))
    for k, v in _1.items():
        ax.plot(numpy.arange(0, len(games) + 1, 1), v, label=k)
        ax.grid(True)
        ax.legend()
    scores = {k: [0] for k in players.keys()}
    fig, ax = matplotlib.pyplot.subplots()
    ax.set_title('scores dynamic')
    for g in games:
        for k, v in scores.items():
            v.append(v[-1] + g[k]['score'])
    for k, v in scores.items():
        ax.plot(numpy.arange(0, len(games) + 1, 1), v, label=k)
        ax.grid(True)
        ax.legend()
    fig, ax = matplotlib.pyplot.subplots()
    ax.set_title('scores diffs')
    bins = numpy.arange(
        min(min(v['score_diffs']) for v in players.values()),
        max(max(v['score_diffs']) for v in players.values()) + 1
    )
    for k, v in players.items():
        ax.hist(v['score_diffs'], bins=bins, label=k, alpha=0.5)
        ax.set_xticks(bins)
        ax.grid(True)
        ax.legend()
    matplotlib.pyplot.show()


def ratio(values):
    if values[1] == values[0]:
        return 1
    elif values[0] == 0:
        return float('inf')
    else:
        return values[1] / values[0]


def row(*args):
    print(('{:>25}' * len(args)).format(*args))


def collect_data(paths):
    for path in paths:
        content = read_result(path)
        if content:
            yield parse_result(content)


def read_result(path):
    with open(path) as f:
        return f.read()


def parse_result(content):
    first, second, seed = content.split('\n')[:3]
    return dict(first=parse_record(first), second=parse_record(second), seed=seed)


def parse_record(content):
    place, score, status = content.split(':')[:3]
    return dict(place=int(place), score=int(score), status=status.strip())


if __name__ == '__main__':
    main()