Python 列表使用技巧

cutercorley

发布于 2020-07-23 16:41:24

1.1K0

文章被收录于专栏：Corley的开发笔记Corley的开发笔记

文章目录

1.列表表达式与列表排序
2.按照子列表中的某个元素对列表进行排序

1.列表表达式与列表排序

列表中的元素也是可迭代的对象如列表、元组等时，要根据这些元素的某个子元素对列表排序，常规排序方式失效，需要用sorted()函数并指定key。题目：

输入一组数到列表nums，请找到列表中任意两个元素相加能够等于9的元素，形成一个元组，使其小数在前大数在后，如：(2,7),(1,8)。重复的元组元素只保留一个，结果按元组第一个元素从小到大顺序输出。【样例输入】 numbers:3,4,5,7,2,8,1 【样例输出】 [(1, 8), (2, 7), (4, 5)]

示例解答如下：

def get_tuple(num_list):
    temp_list = []
    for i in num_list:
        if (9 - i) in num_list:
            min = (9 - i) if (i >= (9 - i)) else i
            max = i if min == (9 - i) else (9 - i)
            if (min,max) not in temp_list:
                temp_list.append((min,max))
    return temp_list

nums = input()
num_list = [int(i) for i in nums.split(',')]

result_list = get_tuple(num_list)
result_list = sorted(result_list,key=lambda i:i[0])

print(result_list)

代码运行演示如下：

说明： [int(i) for i in nums.split(',')]是列表表达式，有更高的执行效率； sorted(result_list,key=lambda i:i[0])对列表进行排序，因为列表中有多个元素，要根据每个元素的第一个子元素来排序，必须使用key来指定排序所依据的元素，同时还是用了lambda表达式，给定一个元素，返回第一个子元素。

2.按照子列表中的某个元素对列表进行排序

例如，有一个列表unsorted_list = [['a', 'b', 'c', 5, 'd'], ['e', 'f', 'g', 3, 'h'], ['i', 'j', 'k', 4, 'm']]，需要按照每个子列表中的第4个元素（即整数）对列表进行排序，有3种方式：方式一——使用lambda表达式

unsorted_list = [['a', 'b', 'c', 5, 'd'], ['e', 'f', 'g', 3, 'h'], ['i', 'j', 'k', 4, 'm']]
sorted_list = sorted(unsorted_list, key=lambda x: x[3])
print(unsorted_list)
print(sorted_list)

打印：

[['a', 'b', 'c', 5, 'd'], ['e', 'f', 'g', 3, 'h'], ['i', 'j', 'k', 4, 'm']]
[['e', 'f', 'g', 3, 'h'], ['i', 'j', 'k', 4, 'm'], ['a', 'b', 'c', 5, 'd']]

显然，得到了排序后的新列表；其中，lambda表达式返回每个子列表的第4个元素，sorted函数使用lambda表达式的返回值作为key来对列表排序。

方式二——使用itemgetter类

from operator import itemgetter
unsorted_list = [['a', 'b', 'c', 5, 'd'], ['e', 'f', 'g', 3, 'h'], ['i', 'j', 'k', 4, 'm']]
sorted_list = sorted(unsorted_list, key=itemgetter(3), reverse=True)
print(unsorted_list)
print(sorted_list)

打印：

[['a', 'b', 'c', 5, 'd'], ['e', 'f', 'g', 3, 'h'], ['i', 'j', 'k', 4, 'm']]
[['a', 'b', 'c', 5, 'd'], ['i', 'j', 'k', 4, 'm'], ['e', 'f', 'g', 3, 'h']]

显然，得到了排序的结果；可以用reverse参数使结果倒序。

方式三——调用列表的sort()方法

unsorted_list = [['a', 'b', 'c', 5, 'd'], ['e', 'f', 'g', 3, 'h'], ['i', 'j', 'k', 4, 'm']]
print(unsorted_list)
unsorted_list.sort(key=lambda x: x[3])
print(unsorted_list)

显然，得到了正确的排序结果；调用sort()方法是对原列表排序，不会产生新的列表。

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2020/04/12 ，如有侵权请联系 cloudcommunity@tencent.com 删除

serverless

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

serverless

登录后参与评论

0 条评论

热度