Python Lambda If Function With 2