Get Array From User In Python Program