Extract list of attributes from list of objects in python

Question:

I have an uniform list of objects in python:

class myClass(object):
    def __init__(self, attr):
        self.attr = attr
        self.other = None

objs = [myClass (i) for i in range(10)]

Now I want to extract a list with some attribute of that class (let’s say attr), in order to pass it so some function (for plotting that data for example)

What is the pythonic way of doing it,

attr=[o.attr for o in objsm]

?

Maybe derive list and add a method to it, so I can use some idiom like

objs.getattribute("attr")

?

Asked By: franjesus

||

Answers:

attrs = [o.attr for o in objs] was the right code for making a list like the one you describe. Don’t try to subclass list for this. Is there something you did not like about that snippet?

Answered By: Mike Graham

You can also write:

attr=(o.attr for o in objsm)

This way you get a generator that conserves memory. For more benefits look at Generator Expressions.

Answered By: Reto Aebersold
Categories: questions Tags: , ,
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.