Python Code For Array Input From User