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

如何使用typing.io在Python中读取和写入CSV文件

发布时间:2024-01-19 02:20:15

在Python中读取和写入CSV文件可以使用csv模块,该模块提供了方便的方法来处理CSV格式的文件。下面是一个使用csv模块读取和写入CSV文件的例子。

首先,需要导入csv模块:

import csv

### 读取CSV文件

要读取CSV文件,可以使用csv.reader函数。这个函数需要接收一个打开的CSV文件对象作为参数,并返回一个reader对象,可以迭代读取CSV文件的每一行。以下是一个读取CSV文件的例子:

with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        # 操作每一行的数据
        print(row)

上述代码假设有一个名为data.csv的CSV文件,reader对象可以迭代读取文件的每一行,并将每一行解析为一个列表(每个元素对应文件中的一个字段)。

### 写入CSV文件

要写入CSV文件,可以使用csv.writer函数。这个函数需要接收一个打开的CSV文件对象和一个可选的dialect参数,返回一个writer对象,可以使用writerow方法写入一行CSV数据。以下是一个写入CSV文件的例子:

data = [
    ['Name', 'Age', 'Gender'],
    ['John', '25', 'Male'],
    ['Jane', '30', 'Female']
]

with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

上述代码将一个名为data.csv的CSV文件创建到当前目录中,并将列表data的每个子列表写入作为每一行的数据。

### 自定义分隔符

默认情况下,csv.readercsv.writer使用逗号作为字段的分隔符。如果CSV文件中的字段分隔符不是逗号,可以使用csv.reader(file, delimiter='分隔符')csv.writer(file, delimiter='分隔符')来自定义分隔符。

例如,如果CSV文件使用制表符作为字段的分隔符,可以这样读取和写入CSV文件:

with open('data.csv', 'r') as file:
    reader = csv.reader(file, delimiter='\t')
    for row in reader:
        # 操作每一行的数据
        print(row)

with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file, delimiter='\t')
    writer.writerows(data)

上述代码中,delimiter='\t'将制表符作为字段的分隔符。

以上就是使用csv模块读取和写入CSV文件的简单例子。希望对你有所帮助!