The article linked above was a good starting ground but for the amount of processing I needed to it was by no means fast enough or efficient enough, so I rewrote a bunch of it and put the source online here - https://github.com/kura/ssh-attack-visualisation/
It is all written in Python with heavy usage of Numpy.