How To Take Input For Array In Python