How to create NBA shot charts in Python