How To Get Array Values From User In Python