Truncated Distribution — Interview Query python problem