plot.py

import statistics

import seaborn as sns
import matplotlib.pyplot as plt


def stats(values, plot=False):
    # statistics.mode throws an error if there are multiple modes
    try:
        mode = statistics.mode(values)
    except:
        mode = None

    d = {
        "len": len(values),
        "max": max(values),
        "min": min(values),
        "mean": statistics.mean(values),
        "median": statistics.median(values),
        "mode": mode,
        "stdev": statistics.stdev(values),
    }

    if plot:
        sns.distplot(values)
        plt.show()

    return d