User Array Input In Python Definition Function