python 多维数组的排序

发布时间：2019-09-03 08:55:52编辑：auto阅读（1889）

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



关键字：

上一篇： python mock 的一些小例子

下一篇： SMOTE python实现