Python Assign Variable Based Model