我有一份csv数据,我只想要最后三列数据,并输出新文件,求代码
时间: 2023-12-20 07:03:15 浏览: 76
用python给csv里的数据排序的具体代码
假设你的CSV文件名为`data.csv`,数据以逗号分隔,每行数据有N列,你只需要最后三列,然后输出到一个新文件`new_data.csv`中,可以使用以下Python代码:
```python
import csv
# 打开原始文件
with open('data.csv', 'r') as f:
reader = csv.reader(f)
# 打开新文件
with open('new_data.csv', 'w', newline='') as new_f:
writer = csv.writer(new_f)
# 遍历原始文件每一行数据
for row in reader:
# 只取最后三列数据
new_row = row[-3:]
# 写入新文件
writer.writerow(new_row)
```
以上代码会将最后三列数据提取出来,并写入到新文件`new_data.csv`中。注意要使用`newline=''`参数来避免写入文件时出现空行。
阅读全文