Get Array From User In Python Iteration