How to Take Array Input in a Single Line in Python