python assignment in lambda