Python基于csv模块实现读取与写入csv数据的方法

### Python基于csv模块实现读取与写入csv数据的方法 #### 概述 在Python编程中，处理CSV（逗号分隔值）文件是一项常见任务。CSV文件因其简单且易于理解的特点，在数据交换和存储方面得到了广泛应用。Python的`csv`模块提供了强大的工具来读取和写入CSV文件，使得这一过程变得高效且简洁。 #### CSV文件介绍 CSV文件是一种常用的数据存储格式，用于存储表格数据。每一行通常代表一条记录，而每一列则代表不同的字段或属性。例如，一个包含员工信息的CSV文件可能会有以下列：`org_id`（组织ID）、`org_name`（组织名称）、`state`（状态）和`emp_id`（员工ID）。 #### 使用Python的csv模块 Python标准库中的`csv`模块提供了读取和写入CSV文件的功能。该模块支持多种分隔符，并能自动处理特殊字符如引号等。 #### 读取CSV数据 1. **导入csv模块** ```python import csv ``` 2. **打开文件** 使用`open()`函数以只读模式(`'r'`)打开文件： ```python with open('file.csv', 'r') as file: ``` 3. **创建csv阅读器** 使用`csv.reader()`方法创建一个CSV阅读器对象： ```python reader = csv.reader(file) ``` 4. **遍历CSV文件** 可以通过迭代阅读器对象来读取文件中的每一行： ```python for row in reader: print(row) ``` 5. **示例代码** ```python #-*-coding:gbk-*- import csv, os # 数据文件E:\t.csv sFilename = 'e:' + os.path.sep + 't.csv' # 打开文件 eFile = open(sFilename) # 读取csv文件 eReader = csv.reader(eFile) # 遍历csv对象获取数据，每一条数据都是一个list，每一列是list中的一个元素 # line_num是行号，这里只读取前100行 for row in eReader: if eReader.line_num <= 100: print('行' + str(eReader.line_num) + ':' + str(row)) # 打印行号 else: break # 关闭文件 eFile.close() ``` #### 写入CSV数据 1. **创建csv写入器** ```python writer = csv.writer(file) ``` 2. **写入数据** - `writerow()`: 用于写入单行数据。 - `writerows()`: 用于写入多行数据。 3. **示例代码** ```python #-*-coding:gbk-*- import csv, os sFilename = os.path.join('E:', 'r.csv') # 打开文件以写入模式 eFile = open(sFilename, 'w', newline='') # 创建csv写入器 eWriter = csv.writer(eFile, delimiter=',', lineterminator='\r\n') # 写入表头 eWriter.writerow(['org_id', 'org_name', 'state', 'emp_id']) # 写入数据 eWriter.writerow([123, '销售10', '1', 1111]) eWriter.writerow([1234, '销售16', '0', 11115]) # 关闭文件 eFile.close() ``` #### 其他技巧与注意事项 - **编码问题**：在处理非英文字符时，确保正确设置文件的编码方式，如示例代码中的`-*-coding:gbk-*-*`指定了GBK编码。 - **异常处理**：在实际应用中，应添加适当的异常处理逻辑，以增强代码的健壮性。 - **性能优化**：对于大型文件，考虑使用更高效的内存管理技术，如分块读取。 通过以上介绍，我们可以看到使用Python的`csv`模块来读取和写入CSV文件是非常简单且直观的。无论是简单的数据处理还是复杂的分析需求，`csv`模块都能提供有力的支持。