How To Take Array From User In Python