您的位置:首页 > 编程语言 > Python开发

python 多维数组的排序

2014-04-10 17:15 302 查看
这几天写php程序,发现php里有一个array_multisort()函数十分好用,可以轻松对多维数组进行排序,查了查python的相关资料,视乎没有一个比较直接的函数来完成多维数组的排序

单个数组的排序很简单,直接用sort就能完成,如一下ipython代码:

In [39]: array = [4, 2, 5, 1, 3]
In [40]: array.sort()
In [41]: array
Out[41]: [1, 2, 3, 4, 5]

多维数组的排序如直接用sort讲会按第一维的数据进行排序,如:

In [42]: array = [ ['b', 4], ['e', 2], ['a', 5], ['d', 1], ['c', 3] ]
In [43]: array.sort()
In [44]: array
Out[44]: [ ['a', 5], ['b', 4], ['c', 3], ['d', 1], ['e', 2] ]

如何按第二维的数据进行排序呢,我们可以用sort函数中的key形参,代码接上,如:

In [45]: array.sort(key=lambda x:x[1])#lambda x:x[1]返回list的第二个数据
In [46]: array
Out[46]: [ ['d', 1], ['e', 2], ['c', 3], ['b', 4], ['a', 5] ]

也可以用一个函数实现,比较直接,如:

In [55]: def sort(a):
....: for k in xrange(len(a)):
....: (a[k][0], a[k][1]) = (a[k][1], a[k][0])
....: a.sort()
....: for k in xrange(len(a)):
....: (a[k][0], a[k][1]) = (a[k][1], a[k][0])
....:
In [56]: array = [ ['b', 4], ['e', 2], ['a', 5], ['d', 1], ['c', 3] ]
In [57]: sort(array)
In [58]: array
Out[58]: [ ['d', 1], ['e', 2], ['c', 3], ['b', 4], ['a', 5] ]
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: