python删除csv文件里的数据保留表头
时间: 2024-01-10 09:05:10 浏览: 88
合并所有csv文件并去除表头的python代码.zip
可以使用Python的csv模块和os模块来删除CSV文件中的数据但保留表头。具体步骤如下:
1. 导入csv和os模块:首先,需要导入csv和os模块来处理CSV文件。
```python
import csv
import os
```
2. 打开CSV文件并读取数据:使用csv模块的reader()函数打开CSV文件,并使用next()函数跳过第一行表头。
```python
with open('file.csv', 'r') as f:
reader = csv.reader(f)
header = next(reader)
```
3. 清空CSV文件数据:使用os模块的truncate()函数清空CSV文件中的数据。
```python
with open('file.csv', 'w') as f:
f.truncate(0)
```
4. 写入表头:使用csv模块的writer()函数指定CSV文件写入模式,并将表头写入CSV文件。
```python
with open('file.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerow(header)
```
完整代码如下:
```python
import csv
import os
with open('file.csv', 'r') as f:
reader = csv.reader(f)
header = next(reader)
with open('file.csv', 'w') as f:
f.truncate(0)
with open('file.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerow(header)
```
执行完以上代码后,CSV文件中的数据将会被清空,但是表头仍然保留。
阅读全文