How To Get Array Elements From User In Python