Using class attributes as function parameters?

Question

So I have a working sorting algorithm in Python. (Its exact contents are irrelevant to this question.) It uses a list called 'people' containing class instances, and the function is hard-coded to sort that list by a specific attribute, 'wealth'.

def my_sort(seq):
    # sorts by seq[n].wealth
    ...
my_sort(people)

Now, I'd like to generalize the function so I could sort by any attribute.

def my_sort2(seq, key):
    # sorts by seq[n].key
    ...
my_sort2(people, wealth)

But this, of course, throws an error, because it doesn't know to consider 'wealth' as a class attribute. So, how is this possible to do?

Out of curiosity, what's wrong with the default sorting algorithm in Python? — Katriel
– Katriel, Commented Oct 19, 2011 at 15:24
I have a habit of doing things the hard way when I notice my skills at some area aren't sufficient. — xcel
– xcel, Commented Oct 19, 2011 at 16:07

NPE · Accepted Answer · 2011-10-19 13:54:54Z

10

You could pass the name of the attribute:

def my_sort2(seq, keyname):
    sort by getattr(seq[n], keyname)

my_sort2(people, 'wealth')

or a getter functor:

def my_sort2(seq, keyfunc):
    sort by keyfunc(seq[n])

my_sort2(people, operator.attrgetter('wealth'))

I prefer the latter approach as it is more generic. For example, it easily allows for computed keys.

edited Oct 19, 2011 at 13:54

answered Oct 19, 2011 at 13:45

NPE

503k114 gold badges970 silver badges1k bronze badges

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

Comments

Constantinius · Accepted Answer · 2011-10-19 13:46:12Z

1

The generic attribute getter function getattr should work:

gettattr(obj, name)

answered Oct 19, 2011 at 13:46

Constantinius

35.2k9 gold badges79 silver badges85 bronze badges

Collectives™ on Stack Overflow

Using class attributes as function parameters?

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related