Take Input In Array In Python