python 多维数组的排序

2023-01-31 04:01:55 数组 排序 多维

这几天写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] ]



相关文章