用python读取csv文件中间几行的数据并导入新的csv文件中
时间: 2023-10-05 22:12:58 浏览: 104
可以使用Python的csv模块来读取和写入csv文件。以下是一个示例代码,可以读取csv文件中的中间几行数据,并将其写入新的csv文件中。
```python
import csv
# 设置输入和输出文件名
input_file = 'input.csv'
output_file = 'output.csv'
# 设置要读取的行号范围
start_row = 2
end_row = 5
# 打开输入和输出文件
with open(input_file, 'r') as f_input, open(output_file, 'w', newline='') as f_output:
# 创建csv读写器对象
reader = csv.reader(f_input)
writer = csv.writer(f_output)
# 读取输入文件的前start_row-1行数据
for row_num in range(start_row - 1):
next(reader)
# 读取输入文件的中间行数据并写入输出文件
for row in reader:
if reader.line_num <= end_row:
writer.writerow(row)
# 如果已经读取到了end_row行,则退出循环
if reader.line_num > end_row:
break
```
这个代码片段假设输入文件中有至少start_row行数据,并且end_row不超过文件中的行数。如果需要,可以根据实际情况进行修改。
阅读全文