Array Input Using For Loop In Python 3