Python文件操作函数汇总：读写文本，处理CSV文件

发布时间：2023-06-16 12:09:36

Python是一种非常流行的程序设计语言，而文件操作是在编程中不可避免的需求。Python提供了许多文件操作函数，帮助开发者进行各种读写与处理操作。在本文中，我们将介绍常见的文件操作函数，并详细讨论如何处理CSV文件。

## 读写文本文件

在Python中，读写文本文件非常简单。可以使用内置的open()函数打开文件，进行读取、写入和追加等操作。下面是一些最常用的文件操作函数：

### 打开文件

在Python中，可以使用open()函数来打开文件。

open(file, mode='r', buffering=-1, encoding=None, errors=None, newline=None, closefd=True, opener=None)

函数的参数如下：

* file - 必须是字符串，文件路径及名称。

* mode - 文件打开模式（只读、写入、追加等），默认为只读模式。

* buffering - 缓存大小设置，默认为-1，即使用默认系统缓存。

* encoding - 使用指定的字符集打开文件。

* errors - 指定字符集解码错误处理方式。

* newline - 控制行结束方式。

* closefd - 是否关闭文件句柄。

* opener - 自定义的打开方式。

### 读取文件

当文件被打开后，就可以使用read()函数从文件中读取内容。例如：

f = open("test.txt", "r")
print(f.read())
f.close()

当然，还可以使用readline()函数逐行读取文件内容，如下所示：

f = open("test.txt", "r")
for line in f:
    print(line)
f.close()

### 写入文件

如果需要写入新的内容到文件中，可以使用write()函数。例如：

f = open("test.txt", "w")
f.write("Hello World")
f.close()

请注意，使用写入模式打开文件会清空文件中所有的内容，并在写入新内容后关闭文件。

### 追加文件

如果需要在已有文件的末尾添加新的内容，可以使用追加模式打开文件，然后使用write()函数写入内容。

例如：

f = open("test.txt", "a")
f.write("Hello World")
f.close()

在追加模式下打开文件时，如果文件不存在，Python将创建一个新的文件。注意，所有写入的内容都将添加到文件末尾。

## 处理CSV文件

CSV（Comma-Separated Values）是一种常见的文件格式，用于存储结构化数据。CSV文件由逗号(或其他分隔符)分隔的行组成，每行代表一个数据行。在Python中，处理CSV文件是非常常见的需求，这里我们将介绍如何使用Python处理CSV文件。

在Python中，可以使用csv模块操作CSV文件。该模块实现了CSV文件的读取、写入、解析和操作等功能。

### 读取CSV文件

使用csv模块读取CSV文件非常简单，可以使用reader()函数读取CSV文件，并将其转换为Python中的列表。例如：

import csv

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

以上代码读取data.csv文件并打印其内容。请注意，在打开文件之后，我们将其传递给csv.reader()函数。此函数创建一个CSV数据读取器，将文件转换为Python中的可迭代对象，可以用于逐行获取CSV数据。

### 写入CSV文件

使用csv模块写入CSV文件也非常简单，可以使用writer()函数创建一个CSV数据写入器。这个函数接收一个文件对象，然后将数据写入到文件中。

例如：

import csv

with open('data.csv', 'w') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['Name', 'Age', 'Country'])
    writer.writerow(['John', '25', 'USA'])
    writer.writerow(['Marry', '27', 'China'])

以上代码创建一个data.csv文件，并写入了三行数据。请注意，每行使用了writer.writerow()函数来写入这些数据。此函数可以接受一个列表，并将列表中的数据写入到CSV文件中的一行中。

### 解析CSV文件

有时候，CSV文件中的数据可能包含了一些特定的符号或字符集。在这种情况下，可以使用csv模块的额外参数来解析CSV文件中的内容。

例如，假设我们有一个CSV文件包含了如下内容：

"Name","Age","Country"
"John, Smith","25","USA"
"Marry Horr","27","China"

有人可能会注意到，"Name"字段中的值包含了逗号，这会导致CSV文件解析出现问题。为了避免此问题，可以使用csv模块的额外参数，这些参数可以干以下的事情：

* delimiter - 定义字段之间的分隔符。

* quotechar - 定义包含特殊字符的值所使用的特殊符号。

* escapechar - 定义在两个带有特殊字符的值之间的转义字符。

例如，可以使用以下代码解析上述CSV文件：

import csv

with open('data.csv', 'r') as csvfile:
    reader = csv.reader(csvfile, delimiter=',', quotechar='"')
    for row in reader:
        print(row)

查看以上代码，你会发现，我们已经使用额外的参数告诉CSV读取器如何解析数据。在这种情况下，我们使用逗号作为字段的分隔符，并使用“”作为特殊符号。

### 处理CSV文件中的字典

在读取CSV文件时，有时候会发现将文件数据转换成字典比较方便。幸运的是，Python提供了一个叫做DictReader的CSV数据读取器，可以将行转换为字典。DictWriter也可以用于将字典转换为行。

例如：

import csv

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

以上代码将data.csv文件转换为字典，并输出其内容。在这种情况下，csv.DictReader()函数将首行的值用作字段，然后使用每一行数据创建一个字典。

## 总结

Python提供了许多在读取和创建文件时非常有用的工具。在本文中，我们介绍了一些最常用的文件操作函数，以及一些用于处理CSV文件的功能。

对于处理CSV文件来说，我们可以使用csv模块轻松解析和操作CSV文件。使用Python处理CSV文件非常便捷，而且非常常见。因此，记住这些技巧有助于让你的Python编程更具高效性和可重用性。