使用pyhon实现将信息写入csv文件的功能,要求如下: 1.读入csv文件路径,如果该文件不存在就新建一个csv文件,并在左上角第一格中写入“图名” 2.判断csv文件中第一行的哪一列是第一个没有数据的位置,在该位置中写入“模型名”,并记录该列是第几列
时间: 2024-03-03 12:47:45 浏览: 57
rbm _RBM代码_researchwyi_rbm代码思路_rbm_pyhon_
可以使用 Python 标准库中的 `csv` 模块来实现将信息写入 CSV 文件的功能。以下是实现上述要求的代码:
```python
import csv
import os
def write_csv_file(file_path, model_name):
# 判断文件是否存在
if os.path.exists(file_path):
# 如果文件存在,则打开文件并读取第一行
with open(file_path, 'r', newline='') as f:
reader = csv.reader(f)
header = next(reader)
else:
# 如果文件不存在,则创建文件并写入第一行
with open(file_path, 'w', newline='') as f:
writer = csv.writer(f)
header = ['图名']
writer.writerow(header)
# 判断哪一列是第一个没有数据的位置
with open(file_path, 'a', newline='') as f:
writer = csv.writer(f)
if header:
col_index = len(header)
for i in range(len(header)):
if not header[i]:
col_index = i
break
header[col_index] = '模型名'
writer.writerow(header)
else:
col_index = 0
header = ['模型名']
writer.writerow(header)
# 在指定位置写入模型名
row = [model_name]
for i in range(col_index):
row.append('')
writer.writerow(row)
```
调用该函数时,需要传入两个参数:文件路径和模型名。例如:
```python
write_csv_file('data.csv', 'Model A')
```
这将在 `data.csv` 文件中写入一行,包含图名和模型名信息。如果文件不存在,则会创建一个新的文件,并在左上角写入“图名”。如果文件中已经有数据,则会在第一个没有数据的位置中写入“模型名”,并记录该列是第几列。
注意,上述代码中使用了 `os` 模块来判断文件是否存在,并使用了 `newline=''` 参数来避免写入 CSV 文件时出现空行的问题。
阅读全文