How To Take Input Array In Python