Get Array From User In Python For Loops