Get Array From User In Python For Loop