欢迎访问宙启技术站
智能推送

Python中的csv模块中的函数如何读写CSV文件?

发布时间:2023-05-27 16:39:18

CSV(Comma Separated Values)是一种常见的文件格式,它将数据以逗号分隔的方式存储在文件中。许多应用程序都使用CSV文件进行数据交换,因为它们易于使用和理解。

在Python中,有一个csv模块可以使读写CSV文件更加容易。这个模块提供了一些函数,可以读取和写入CSV文件。以下是一些常用的函数:

1. csv.reader(file, dialect = 'excel',** fmtparams ): 该函数接受一个打开的文件作为参数,并返回一个csv.reader对象。 可以使用它从CSV文件中读取数据。 dialect参数是可选的,可以指定读取文件时使用的分隔符和引用字符。 可选参数fmtparams可以用于指定其他格式化选项。默认情况下,读取器使用Excel方言。

下面是一个简单的示例,它从example.csv文件中读取数据:

import csv  

with open("example.csv","r") as file:    

   reader = csv.reader(file)  

   for row in reader:  

       print(row)

2. csv.writer(file,dialect ='excel',** fmtparams):该函数接受一个打开的文件和一个可选的方言参数,并返回一个csv.writer对象。可以使用它将数据写入CSV文件。 可选参数fmtparams可以用于指定其他格式化选项。默认情况下,写入器使用Excel方言。

以下是一个简单的示例,它将数据写入example.csv文件:

import csv  

with open("example.csv","w") as file:    

   writer = csv.writer(file)   

   writer.writerow(["Name", "Country"])    

   writer.writerow(["John", "USA"])  

   writer.writerow(["Tom", "UK"])

3. csv.DictReader(file,fieldnames = none,restkey = none,restval = none,dialect = 'excel',* * fmtparams):该函数接受一个打开的文件和一个可选的方言参数,并返回一个csv.DictReader对象。 与csv.reader对象不同,DictReader对象返回一个有序字典而不是列表。 字典的键是字段名称,字典的值是单元格的值。 fieldnames参数是可选的,它指定CSV文件中每列的标题。 如果不指定fieldnames,则 行被视为标题。 restkey和restval参数用于指定哪些值存储在键名为restkey的字典中,这些值没有对应的键名。 默认情况下,使用Excel方言。

以下是一个简单的示例,它从example.csv文件中读取数据,并使用DictReader对象返回一个有序字典:

import csv  

with open("example.csv","r") as file:    

   reader = csv.DictReader(file)  

   for row in reader:  

       print(row)

4. csv.DictWriter(file,fieldnames,dialect = 'excel',* * fmtparams):该函数接受一个打开的文件、要写入的列标题列表和一个可选的方言参数。 它返回一个csv.DictWriter对象,它能够将给定的字典列表写入CSV文件。 字典的键必须是给定的列标题之一。默认情况下,使用Excel方言。

以下是一个简单的示例,它将数据写入example.csv文件,并使用DictWriter对象将字典列表写入CSV文件:

import csv  

with open("example.csv","w") as file:    

   writer = csv.DictWriter(file, fieldnames=["Name", "Country"])   

   writer.writeheader()    

   writer.writerow({"Name": "John", "Country": "USA"})  

   writer.writerow({"Name": "Tom", "Country": "UK"})

总结

上面介绍了csv模块中的一些读写CSV文件的常用函数。除了上面的函数外,还有其他一些有用的函数可以在处理CSV文件时使用。无论你是在读取还是写入CSV文件,csv模块都可以使这些任务更加容易。Python的csv模块非常适合用于处理简单的表格数据,比如电子表格、数据表等。这些函数提供了一种快速和简便的方法来处理CSV文件,并能够处理各种数据类型,包括字符串、数字和日期。