Python Assign Variable To Function