Python-将函数分配给变量

Question

I have a list of tuples

student_tuples = [
    ('john', 'A', 15),
    ('jane', 'B', 12),
    ('dave', 'B', 10),
]

I've been trying different ways to sort this, using itemgetter and lambda functions. Sorting by two indices of the tuples can be done with itemgetting and the lambda function, but it must return a tuple. I can't seem to find that anywhere in the documentation that the key function works on tuples.

Anyway, I wanted to know what itemgetter() actually returns, so this works (copied from the itemgetter documentation):

f = itemgetter(1)
print f(student_tuples[0])
----->A

Is there any way to do this WITHOUT having to reassign itemgetter to a variable? It looks like two arguments are being passed, but something like

print itemgetter(1, student_tuples[0])
-----><operator.itemgetter object at 0xf7309c8c>

doesn't give me anything useful.

I'm just fiddling around trying to learn Python and this is confusing me. I don't know where in itemgetter student_tuples[0] is being added as an argument.

Answer 1

The return value of itemgetter(1) is a function (actually, a callable object, but it's used like a function).

The function it returns is roughly equivalent to the function that results from the expression:

lambda x: x[1]

student_tuples[0] isn't added as an argument anywhere in itemgetter . It is passed as an argument to the function-that-was-returned when you call f(student_tuples[0]) .

Since f is the result of itemgetter(1) , it follows that you can do this in one line as:

itemgetter(1)(student_tuples[0])

Answer 2

You need to call the return value of itemgetter() again, not pass in the list as the second argument. Example -

itemgetter(1)(student_tuples[0])

As can be seen from the documentation -

operator.itemgetter(*items)

Return a callable object that fetches item from its operand using the operand's __getitem__() method.

itemgetter() returns a function, which you can again call passing in the actual iterable, to get the value from that particular index in that iterable.

When you pass in multiple values to itemgetter() , it still returns a function, and calling that function would try to get the elements from the iterable using the index you passed in to itemgetter() initially as a tuple. Example -

>>> l =[1,2,3,4,5]
>>> operator.itemgetter(1,2,4)(l)
(2, 3, 5)