Python Script Take Input Array In Python