How to write in more pythonic way?

Question

I want to write this code in more pythonic way. Have you any idea how I can do this?

def counter():
    for x in self.get_xs():
        total_x = 0
        result = (re.sub('[^0-9]', '', x))
        for number in result:
            total_x += int(number)
        yield(total_x)

Answer 1

You can calculate the sum using sum() and a generator expression. If the iterable passed to sum() is empty(ie re.sub() returned '' ) then it will simply return the default start value 0.

def counter():
    r = re.compile('[^0-9]')
    for x in self.get_xs():
        yield sum(int(number) for number in r.sub('', x))

In Python 3.3+ you can use yield from :

def counter():
    r = re.compile('[^0-9]')
    yield from (sum(map(int, r.sub('', x))) for x in self.get_xs())