如何使用 Python 对 CSV 文件中的指定列数据进行排序?
今天将给大家带来,感兴趣的朋友请继续看下去吧!以下内容将会涉及到等等知识点,如果你是正在学习文章或者已经是大佬级别了,都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家!
如何按照指定数据对 csv 文件中的行进行排序
本教程将介绍如何使用 python 对 csv 文件中指定列的数据进行排序并写入新文件,无需使用第三方库。
现有一份名为“西北地区省级行政区人口.csv”的文件,内容如下:
陕西,3954 甘肃,2490 青海,594 宁夏,725 新疆,2589
目标是将该文件中的数据按照人口进行排序,并写入一份新文件“西北地区省级行政区人口(按照人口排序).csv”。
-
读取文件并创建列表
首先,使用 with open 语句打开 csv 文件并读取内容到一个列表中:
with open("西北地区省级行政区人口.csv", mode='r', encoding='u8') as f: data = [line.replace('\n', '') for line in f.readlines()]
这部分代码用 u8 编码打开文件,以逐行读取内容。我们还使用 replace(‘n’, ”) 来移除每一行的换行符。
-
排序列表
要对数据进行排序,可以使用 sort 函数。指定 key 参数以自定义排序规则:
data.sort(key=lambda row: int(row.split(',')[-1]))
这部分代码将每一行分割为一个数组,并将最后一项(即人口数)转换为整数。然后使用这个值作为排序依据。
-
写入新文件
最后,使用 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))
以上就是《如何使用 Python 对 CSV 文件中的指定列数据进行排序?》的详细内容,更多关于的资料请关注公众号!