python--lambda表达式在sort函数中的使用

Posted 2022-09-05 lovewhale1997

1.lambda表达式一般用法

语法:

lamda argument:expression

example:

add = lambda x, y: x+y
print(add(10, 20))
>>> 30

2.lambda表达式在sort函数中的使用

假如a是一个由元组构成的列表，对该列表进行排序时，我们需要用到参数key，也就是关键词，如下面代码所示，lambda是一个匿名函数，是固定写法；x表示匿名函数的输入，即列表中的一个元素，在这里，表示一个元组，x只是临时起的一个名字，你可以使用任意的名字；x[0]表示匿名函数的输出，即元组里的第一个元素，即key = x[0]；所以这句命令的意思就是按照列表中第一个元素进行排序。

a = [(‘b‘, 4), (‘a‘, 12), (‘d‘, 7), (‘h‘, 6), (‘j‘, 3)]
a.sort(key=lambda x: x[0])
print(a)
>>>[(‘a‘, 12), (‘b‘, 4), (‘d‘, 7), (‘h‘, 6), (‘j‘, 3)]

3.举例，leetcode中的一道题

937.重新排列日志

 

你有一个日志数组 logs。每条日志都是以空格分隔的字串。

 

对于每条日志，其第一个字为字母数字标识符。然后，要么：

 

标识符后面的每个字将仅由小写字母组成，或；
标识符后面的每个字将仅由数字组成。
我们将这两种日志分别称为字母日志和数字日志。保证每个日志在其标识符后面至少有一个字。

 

将日志重新排序，使得所有字母日志都排在数字日志之前。字母日志按内容字母顺序排序，忽略标识符；在内容相同时，按标识符排序。数字日志应该按原来的顺序排列。

 

返回日志的最终顺序。

示例 ：

输入：["a1 9 2 3 1","g1 act car","zo4 4 7","ab1 off key dog","a8 act zoo"]
输出：["g1 act car","a8 act zoo","ab1 off key dog","a1 9 2 3 1","zo4 4 7"]

思路：先将日志数组logs分成两个日志数组，一个存储字母日志，一个存储数字日志，根据题意，需将所有字母日志排在数字日志前面，字母日志按内容字母排序，故仅需对字母日志数组排序，最后返回两个数组想家的结果就OK了。

代码如下：

class Solution(object):
    def reorderLogFiles(self, logs):
        """
        :type logs: List[str]
        :rtype: List[str]
        """
        l1=[]
        l2=[]
        for l in logs:
            if l[-1].isalpha():
                l1.append(l)
            else:
                l2.append(l)
        l1.sort(key=lambda x:(x[x.index(‘ ‘)+1:],x[:x.index(‘ ‘)]))
        return l1+l2