How To Set Input Array In Python For Loop