博客

Python操作CSV格式文件的方法大全

  • Post category:Python

下面是一份“Python操作CSV格式文件的方法大全”的完整实例教程。

1. CSV格式文件简介

CSV格式是指逗号分隔符文件（Comma-separated values），是一种常见的电子表格文件格式，也是简单易用的数据交换格式。CSV格式文件中的数据以行为单位，每行数据由逗号分隔。通常第一行是表头，用于描述每一列的含义。

2. 读取CSV格式文件的方法

Python中读取CSV格式文件可以使用csv模块。csv模块提供了DictReader和reader两个函数，分别用于读取具有头部信息的CSV格式文件和不带头部信息的CSV格式文件。

2.1 使用DictReader函数读取CSV格式文件

以下是一个示例代码，展示了如何使用DictReader函数读取具有头部信息的CSV格式文件。

import csv

with open('data.csv', newline='') as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        print(row['name'], row['age'])

在上面的代码中，通过csv模块的DictReader函数读取data.csv文件中的数据，并以字典形式返回每一行数据。通过遍历字典，可以获取每一行数据中指定列的值。

2.2 使用reader函数读取CSV格式文件

以下是一个示例代码，展示了如何使用reader函数读取不带头部信息的CSV格式文件。

import csv

with open('data.csv', newline='') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

在上面的代码中，通过csv模块的reader函数读取data.csv文件中的数据，返回每一行数据组成的列表。通过遍历列表，可以获取每个单元格的值。

3. 写入CSV格式文件的方法

Python中写入CSV格式文件可以使用csv模块中的writer和DictWriter函数，分别用于写入不带头部信息的CSV格式文件和带头部信息的CSV格式文件。

3.1 使用writer函数写入CSV格式文件

以下是一个示例代码，展示了如何使用writer函数写入不带头部信息的CSV格式文件。

import csv

data = [['name', 'age'], ['Tom', 18], ['Jerry', 20]]

with open('data.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    for row in data:
        writer.writerow(row)

在上面的代码中，通过在列表中嵌套列表的形式将数据存放在data变量中，并通过csv模块的writer函数将数据写入data.csv文件中。

3.2 使用DictWriter函数写入CSV格式文件

以下是一个示例代码，展示了如何使用DictWriter函数写入带头部信息的CSV格式文件。

import csv

data = [{'name': 'Tom', 'age': 18}, {'name': 'Jerry', 'age': 20}]

with open('data.csv', 'w', newline='') as csvfile:
    fieldnames = ['name', 'age']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
    writer.writeheader()
    for row in data:
        writer.writerow(row)

在上面的代码中，通过在字典中嵌套字典的形式将数据存放在data变量中，并通过csv模块的DictWriter函数将数据写入data.csv文件中。通过传递fieldnames参数指定每一列的表头信息。写入数据之前，需要先调用writeheader方法写入表头信息。

以上就是Python操作CSV格式文件的方法大全的完整实例教程。

Tags: ,