How To Take Array Input In Python In One Line