How To Take Array User Input In Python