Python Code For Array Input