Python CSV 超简明用法

时间：2023-03-08 16:56:50

平常经常会用CSV存储数据，不可避免的会跟CSV文件的读写操作扯上关系。

Python有CSV这个Package来解决这个问题，官网也有比较详细的教程 https://docs.python.org/3/library/csv.html 。但是我觉得往往大家只是想进行简单的读写、修改等操作，网上纷纷扰扰的教程和官网的教程有点大而全了，反而无法让大家在最短的时间里实现最简单的功能。 所以我写一个简明用法来方便大家。

假设大家有一个comma delimited格式的csv文档(如果只有xlsx格式的Excel文件，可以先用Office Excel的另存为功能将其转为comma delimited格式的csv文档)：

Name, School, Num

Allen, ECNU, 1

Eric, ECNU, 2

Shan, ECNU, 3

通过以下代码进行读取操作：

import csv

def csv_to_list(csv_path):

    print("Begin to extract csv")

    csv_content=[]

    with open(csv_path) as csvfile:

        reader = csv.reader(csvfile)

        for row in reader:

            csv_content.append(row)

    print("The content of CSV file is: \n%s" % csv_content)

    return csv_content

获得的是一个列表：

[ [Name, School, Num], [Allen, ECNU, 1] , [Eric, ECNU, 2], [Shan, ECNU, 3] ]

假如我们想生成一个新的CSV文件，不包含School这一列:

import csv

def write_csv(csv_write_path,csv_content):

    print("Remove the SCHOOL field of the information:")

    for each_line in csv_content:

        del(each_line[1])

    print("Ready to write data to %s" % csv_write_path)

    with open(csv_write_path,'w',newline='') as csvfile:

        writer = csv.writer(csvfile)

        writer.writerows(csv_content)

    print("Content has written to new csv %s" % csv_write_path)

最终的结果会生成这样的一个新CSV：

Name, Num

Allen, 1

Eric, 2

Shan, 3

相关文章