How To Get Array From User In Python