How To Return Integer Array In Python