了解python中的lambda函数

Question

我是 python 新手，我创建了一个脚本来对来自 cisco 路由器的“显示 IP 记帐”信息进行排序。该脚本读取一个文件并将每一行分解为一个列表，然后创建每行的列表。所以我最终得到了一个列表列表：

list a = [[192.168.0.1,172.16.0.1,3434,12222424],[192.168.2.1,172.12.0.1,33334,12667896722424]]

我希望能够按列表中列表的第三列或第四列进行排序。

我可以使用 lambda 函数来做到这一点，但我的问题是如何使用标准函数来复制它？

下面是我的代码：

from sys import argv

script, option, filename = argv
a=[]
b=[]

def openfile(filename):
  file = open(filename)
  for line in file:
    if not line.startswith("  "):
      a.append((line.split()))
  return a

def sort(a,num):
  b = sorted(a, reverse=True, key=lambda x: int(x[num]))
  return b

def top5(b):
  print "Source     Destination Packets     Bytes"
  for i in b[:4]: 
    print i[0]+"    "+i[1]+"    "+i[2]+"        "+i[3]

def main(option):
  a = openfile(filename)
  if option == "--bytes":
    b = sort(a,3)
    top5(b)
  elif option == "--packets":
    b = sort(a,2)
    top5(b)
  else:
    print """
    Not a valid switch, 
    --bytes to sort by bytes 
    --packets to sort by packets."""


main(option)

所以我的问题是如何将 lambda 函数复制为标准的自定义排序函数？我正在尝试弄清楚这是如何工作的。

b = sorted(a, reverse=True, <strong>key=lambda x: int(x[num])</strong>)

Answer 1

如何将 lambda 函数复制为标准的自定义排序函数？

你是这个意思吗：

def sort(a, num):
  def key(x):
    return int(x[num])
  return sorted(a, reverse=True, key=key)

或者这个：

from functools import partial

def key(num, x):
  return int(x[num])

def sort(a, num):
  return sorted(a, reverse=True, key=partial(key, num))

?

Answer 2

Python 提供了operator.itemgetter 来做这种事情：

def sort(a, num):
    return sorted(a, reverse=True, key=operator.itemgetter(num))

编辑

正如@NPE 指出的那样，这不会将密钥转换为int 进行排序。为此，您最好坚持使用 lambda。