charter.py

import json
import datetime

import matplotlib.pyplot as plt
import matplotlib.dates as mdates

colors = [
    "black",
    "xkcd:dark red",
    "red",
    "orange",
    "yellow",
    "xkcd:lime",
    "green",
    "xkcd:dark green",
    "xkcd:cyan",
    "xkcd:blue",
    "xkcd:dark blue",
]


def do_chart():
    with open("wanikani_perf.json") as d:
        data = json.load(d)

    fig = plt.figure(figsize=[8, 13])
    for i, t in enumerate(["radical", "kanji", "vocabulary"]):
        ax = fig.add_subplot(311 + i)
        ax.set_title(t.capitalize())
        labels = []
        for x in range(1, 10):
            f = []
            s = []
            for k, v in data.items():
                f.append(datetime.datetime.fromisoformat(k))
                try:
                    s.append(v[t][f"{x}"])
                except KeyError:
                    s.append(0)
            if not any(a != 0 for a in s):
                continue
            ax.plot(f, s, color=colors[x])
            labels.append(str(x))
        ax.legend(labels, loc="upper left", ncol=4)
        ax.xaxis.set_major_formatter(mdates.DateFormatter('%Y\n%m-%d'))
    plt.show()


if __name__ == '__main__':
    do_chart()