How To Take Input In Python Array