How To Give Array As Input In Python 3