I've included an ipython notebook for anyone who might want to check it out (hopefully it works): http://nbviewer.ipython.org/github/samlagasse/LaGasse_QT/blob/master/QHE_currentdensity.ipynb
In this example its pretty similar (I just think it looks really cool), but in another I did looking a focusing effects in graphene, you really need the current density map.