如何使用 Python 对 CSV 文件中的指定列数据进行排序？

如何按照指定数据对 csv 文件中的行进行排序

本教程将介绍如何使用 python 对 csv 文件中指定列的数据进行排序并写入新文件，无需使用第三方库。

问题描述

现有一份名为“西北地区省级行政区人口.csv”的文件，内容如下：

陕西,3954
甘肃,2490
青海,594
宁夏,725
新疆,2589

目标是将该文件中的数据按照人口进行排序，并写入一份新文件“西北地区省级行政区人口(按照人口排序).csv”。

立即学习“Python免费学习笔记（深入）”；

分步实现

1. 读取文件并创建列表

   首先，使用 with open 语句打开 csv 文件并读取内容到一个列表中：

   

   影谱

   汉语电影AI辅助创作平台

   下载

   with open("西北地区省级行政区人口.csv", mode='r', encoding='u8') as f:
       data = [line.replace('\n', '') for line in f.readlines()]

   这部分代码用 u8 编码打开文件，以逐行读取内容。我们还使用 replace('n', '') 来移除每一行的换行符。

2. 排序列表

   要对数据进行排序，可以使用 sort 函数。指定 key 参数以自定义排序规则：

   data.sort(key=lambda row: int(row.split(',')[-1]))

   这部分代码将每一行分割为一个数组，并将最后一项（即人口数）转换为整数。然后使用这个值作为排序依据。

3. 写入新文件

   最后，使用 with open 语句打开一个新文件，并写入排序后的数据：

   with open("西北地区省级行政区人口(按照人口排序).csv", mode='w', encoding='u8') as f:
       f.write('\n'.join(data))

最终代码

组合上述步骤，得到最终的代码：

with open("西北地区省级行政区人口.csv", mode='r', encoding='u8') as f:
    data = [line.replace('\n', '') for line in f.readlines()]
data.sort(key=lambda row: int(row.split(',')[-1]))
with open("西北地区省级行政区人口(按照人口排序).csv", mode='w', encoding='u8') as f:
    f.write('\n'.join(data))