Python Assign List Values To Variables In Math