Get Array Input Python 3