Python In Line Conditional Distribution