📜  读取csv python(1)

📅  最后修改于: 2023-12-03 15:12:11.371000             🧑  作者: Mango

读取CSV文件的相关介绍

CSV(Comma Separated Values)是一种常见的文件格式,也是数据交换的一种通用方式。在Python中,我们可以使用csv模块来读取和处理CSV文件。csv模块提供了一个DictReader类,它允许我们将CSV文件作为一个Python字典对象来使用,这样我们就可以像处理字典一样来访问CSV文件中的数据了。

CSV文件的读取

我们可以使用csv模块的reader()函数来读取CSV文件。下面是一个读取csv文件的示例代码:

import csv

with open('example.csv') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row)

在本例中,我们首先打开了一个名为example.csv的CSV文件,并将其赋给了一个变量f。接下来,我们创建了一个csv.reader对象并将其赋给了变量reader。接着,我们使用for循环遍历csv文件中的每一行数据,并将其打印出来。上面这段代码将会输出如下内容:

['name', 'age', 'gender']
['Tom', '20', 'M']
['Jane', '18', 'F']
['John', '26', 'M']
['Marry', '21', 'F']
CSV文件的写入

我们可以使用csv模块的writer()函数来写入CSV文件。下面是一个将数据写入csv文件的示例代码:

import csv

rows = [
    ("Tom", "20", "M"),
    ("Jane", "18", "F"),
    ("John", "26", "M"),
    ("Marry", "21", "F")
]

with open('example.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(['name', 'age', 'gender'])
    for row in rows:
        writer.writerow(row)

在本例中,我们首先定义了一个列表rows,其中包含了需要写入CSV文件的数据。接下来,我们打开一个名为example.csv的CSV文件,并将其赋给变量f。我们使用csv.writer()函数创建了一个csv.writer对象,并将其赋给变量writer。然后,我们使用writerow()方法将表头写入文件,再用for循环遍历rows列表写入每一行数据。

CSV文件作为字典对象使用

我们可以使用csv模块的DictReader类,将CSV文件作为Python字典对象来使用。下面是一个使用DictReader来处理CSV文件的示例:

import csv

with open('example.csv') as f:
    reader = csv.DictReader(f)
    for row in reader:
        print(row)

在本例中,我们首先打开了一个名为example.csv的CSV文件,并将其赋给变量f。我们使用csv.DictReader函数创建了一个字典对象,并将其赋给变量reader。然后,我们使用for循环遍历CSV文件中的每一行数据,并将其打印出来。上面这段代码将会输出如下内容:

{'name': 'Tom', 'age': '20', 'gender': 'M'}
{'name': 'Jane', 'age': '18', 'gender': 'F'}
{'name': 'John', 'age': '26', 'gender': 'M'}
{'name': 'Marry', 'age': '21', 'gender': 'F'}

通过DictReader类,我们将CSV文件中的每一行数据都转换为一个包含字段名和值的字典,从而可以很方便地访问和处理CSV文件中的数据。

总结

通过csv模块,我们可以很方便地读取和处理CSV文件。在Python中,CSV文件是非常常见的一种数据交换方式,因此,学会如何使用csv模块对CSV文件进行读写处理是非常有必要的。你可以根据自己的需求使用上述的三种方法来处理CSV文件,这里提供的只是一些最基础的示例代码,你可以根据自己的需求进行更加丰富和复杂的处理。