Writing functions inside classes in Python

Question

I'm new to writing classes in Python and had a quick question. Here is some code below.

class Parse_profile():

  def __init__(self, page):
    self.page = page

  def get_title(self):
    list_of_titles = []
    for title in self.page.find_all(.....):
      list_of_titles.append(title.get_text())
    return list_of_titles

  def get_companies(self):
    list_of_companies = []
    for company in self.page.find_all(.....):
      list_of_companies.append(company)
    return list_of_companies

I want to create a third function that will take in both list_of_companies and list_of_titles (and more later on) and combine them into one list.

How do I do that?

Does this new function need information from the Parse_profile instance? — Chad S.
– Chad S., Commented Dec 17, 2015 at 0:08
Can you explain exactly how you want to combine them into a single list? Just add all the companies to the end of the list of titles? — Tom Dalton
– Tom Dalton, Commented Dec 17, 2015 at 0:08

timgeb · Accepted Answer · 2015-12-17 00:10:38Z

3

def get_companies_and_titles(self):
    return self.get_companies() + self.get_title()

This will get you a list with all companies followed by all titles. In case you want (company, title) tuples in your result-list, use:

def get_companies_and_titles(self):
    return zip(self.get_companies(), self.get_title())

The second option can yield unexpected results if the lists are not of the same length. Have a look at the documentation for zip and izip_longest in that case.

edited Dec 17, 2015 at 0:10

answered Dec 17, 2015 at 0:07

timgeb

79.2k20 gold badges129 silver badges150 bronze badges

Sign up to request clarification or add additional context in comments.

4 Comments

Joran Beasley Over a year ago

I upvoted because this answers OP's question as stated... that said I am guessing this is not actually what OP meant :P

timgeb Over a year ago

@JoranBeasley reading the question again it seems a litte ambiguous. Let's see if the author will clarify.

dropWizard Over a year ago

AHHH! Yes, this is what I was looking for. I was writing get_companies() inside the function and it was throwing an error. self.get_companies() works :-) Thank you!! I wasn't sure how to pass a function into another function inside a class.

timgeb Over a year ago

@MorganAllen no problem, happy coding.

Collectives™ on Stack Overflow

Writing functions inside classes in Python

1 Answer 1

4 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

4 Comments

Your Answer

Sign up or log in

Post as a guest

Related