Getting Array Input In Python Syntax For Loop