Python Lambda Return Value Based