Best Way To Plot State In Python