Sorting dictionary using operator.itemgetter
In [6]: sorted(mydict.iteritems(), key=lambda (k,v): operator.itemgetter(1)(v))
Out[6]:
[('a2', ['e', 2]),
('a4', ['s', 2]),
('a3', ['h', 3]),
('a1', ['g', 6]),
('a6', ['y', 7]),
('a5', ['j', 9])]
The key parameter is always a function that is fed one item from the iterable (mydict.iteritems()
) at a time. In this case, an item could be something like
('a2',['e',2])
So we need a function that can take ('a2',['e',2])
as input and return 2.
lambda (k,v): ...
is an anonymous function which takes one argument -- a 2-tuple -- and unpacks it into k
and v
. So when the lambda
function is applied to our item, k
would be 'a2'
and v
would be ['e',2]
.
lambda (k,v): operator.itemgetter(1)(v)
applied to our item thus returns
operator.itemgetter(1)(['e',2])
, which "itemgets" the second item in ['e',2]
, which is 2.
Note that lambda (k,v): operator.itemgetter(1)(v)
is not a good way to code in Python. As gnibbler points out, operator.itemgetter(1)
is recomputed for each item. That's inefficient. The point of using operator.itemgetter(1)
is to create a function that can be applied many times. You don't want to re-create the function each time. lambda (k,v): v[1]
is more readable, and faster:
In [15]: %timeit sorted(mydict.iteritems(), key=lambda (k,v): v[1])
100000 loops, best of 3: 7.55 us per loop
In [16]: %timeit sorted(mydict.iteritems(), key=lambda (k,v): operator.itemgetter(1)(v))
100000 loops, best of 3: 11.2 us per loop
The answer is -- you can't. operator.itemgetter(i)
returns a callable that returns the item i
of its argument, that is
f = operator.itemgetter(i)
f(d) == d[i]
it will never return simething like d[i][j]
. If you really want to do this in a purely functional style, you can write your own compose()
function:
def compose(f, g):
return lambda *args: f(g(*args))
and use
sorted(mydict.iteritems(), key=compose(operator.itemgetter(1),
operator.itemgetter(1)))
Note that I did not recommend to do this :)