本文共 962 字，大约阅读时间需要 3 分钟。

【Python — lambda表达式】对数据集中的各国家人口数进行排序

下面是关于世界国家人口统计的数据，利用lambda表达式对其进行排序。

数据集以制表符\t为单位进行排版，主要包含以下几列信息：

• 国家名称
• 大洲信息
• 人口数量

数据提取与解析

代码逻辑如下：

import pandas as pd
data = open('data/countries_zh.txt', 'r', encoding='utf-8')
countries = []
for line in data:
    line = line.strip()
    arr = line.split("\t")
    name = arr[1]
    caption = arr[3]
    population = int(arr[4])
    countries.append([name, caption, population])

代码功能：

数据排序方法

以下是两种常见的排序方法：

方法一：自定义函数

def get_population(country):
    return country[2]
countries.sort(key=get_population)
for each_country in countries:
    print(each_country)

代码功能：

方法二：Lambda表达式

countries.sort(key=lambda country: country[2])
for each_country in countries:
    print(each_country)

代码功能：

总结

lambda表达式在排序操作中具有显著优势：

在实际应用中，选择哪种方法取决于具体需求。如果需要复杂的排序逻辑，建议使用自定义函数；而对于简单的场景，lambda表达式是更优选择。

转载地址：http://bweq.baihongyu.com/