How To Set Input Array In Python Code For Snake