Charge Density Distribution Graph Python Code