Getting Array Input In Python Definition