Python Script Take Input Array