Python函数使用：读写CSV文件

发布时间：2023-06-23 10:44:37

Python是一种强大的编程语言，用来开发各种的应用程序，比如Web应用、桌面应用和机器学习等。在Python中，CSV文件是一种非常常见的文件格式，它通常是用于存储和读取数据，类似于Excel表格。本文将讨论如何使用Python函数读写CSV文件。

1. 读取CSV文件

使用Python内置的csv模块可以轻松读取CSV文件。这个模块提供了一个CSVReader对象，可以用于逐行读取CSV文件中的数据。下面是一个示例代码：

import csv

with open('example.csv') as csv_file:

csv_reader = csv.reader(csv_file)

for row in csv_reader:

print(row)

在这个示例中，我们打开一个名为“example.csv”的文件，然后使用csv.reader()函数创建一个CSVReader对象。第二个参数可以用于指定文件中的定界符，例如分隔符是逗号或分号等。在这个例子中，我们忽略这个参数，因为默认情况下，CSV文件中分隔符是逗号。接下来，我们使用一个for循环遍历文件中的每一行，并打印出来。

2. 写入CSV文件

使用Python内置的csv模块也可以轻松写入CSV文件。这个模块提供了一个CSVWriter对象，可以用于在CSV文件中逐行写入数据。下面是一个示例代码：

import csv

with open('example.csv', mode='w') as csv_file:

csv_writer = csv.writer(csv_file)

csv_writer.writerow(['姓名', '年龄', '性别'])

csv_writer.writerow(['张三', '18', '男'])

csv_writer.writerow(['李四', '20', '女'])

在这个示例中，我们打开一个名为“example.csv”的文件，然后使用csv.writer()函数创建一个CSVWriter对象。第二个参数可以用于指定文件中的定界符。在这个例子中，我们忽略这个参数，因为默认情况下，CSV文件中分隔符是逗号。然后，我们使用writerow()函数将三行数据写入到文件中，每一行都用一个列表表示。

3. 处理CSV文件中的缺失值

在实际操作中，CSV文件中可能会存在一些缺失值，这些值通常用空格或空字符串表示。Python内置的csv模块提供了一个解决方案，使用csv.reader对象的skipinitialspace参数可以自动跳过行首的空格。下面是一个示例代码：

import csv

with open('example.csv') as csv_file:

csv_reader = csv.reader(csv_file, skipinitialspace=True)

for row in csv_reader:

print(row)

在这个示例中，我们使用csv.reader对象的skipinitialspace参数来跳过空格。这个参数默认是False，如果设置为True，则会忽略行首的空格。这使得处理CSV文件中的空格变得更加容易。

4. 处理CSV文件中的编码问题

在读取和写入CSV文件时，经常会出现编码问题。Python内置的csv模块同样提供了一些解决方案。例如，可以使用codecs模块中的open()函数来指定编码格式。

下面是一个示例代码：

import csv

import codecs

with codecs.open('example.csv', 'r', encoding='utf-8') as csv_file:

csv_reader = csv.reader(csv_file)

for row in csv_reader:

print(row)

在这个示例中，我们使用codecs.open()函数来打开CSV文件，并指定编码格式为utf-8。然后，我们使用csv.reader()函数创建一个CSVReader对象，并遍历文件中的每一行。这个方法可以帮助我们解决CSV文件中的编码问题。

总结

在Python中，读写CSV文件是一项非常普遍的任务。本文介绍了如何使用Python内置的csv模块来读写CSV文件，并处理CSV文件中的缺失值和编码问题。尽管CSV文件是非常简单的格式，但是在实际应用中，仍需要仔细处理数据，以确保数据的准确性和完整性。