1

我是 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, key=lambda x: int(x[num]))

4

2 回答 2

4

如何将 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))

?

于 2013-01-25T22:41:26.040 回答
1

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

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

编辑

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

于 2013-01-25T22:49:35.507 回答