将Excel文件转换为csv文件,并将表头的id,lable,probability更改为uuid,prediction,proba
时间: 2024-09-08 21:03:07 浏览: 100
要将Excel文件转换为CSV文件并重命名列标题,你可以使用Python的`pandas`库。首先确保已安装了`openpyxl`(用于读取Excel)和`pandas`(用于数据处理)。以下是一段示例代码:
```python
import pandas as pd
from uuid import uuid4
# 读取Excel文件
xls = pd.read_excel('input.xlsx', engine='openpyxl') # 替换 'input.xlsx' 为你实际的Excel文件路径
# 重命名列标题
xls = xls.rename(columns={'id': 'uuid', 'lable': 'prediction', 'probability': 'proba'})
# 将DataFrame保存为CSV文件
xls.to_csv('output.csv', index=False) # 替换 'output.csv' 为你希望输出的CSV文件路径
# 如果你想创建UUID列而不是简单地重命名id,可以在rename之前生成一个新的列:
# xls['uuid'] = xls['id'].apply(lambda x: str(uuid4()))
```
这段代码做了以下几件事:
1. 使用`read_excel`函数从Excel文件加载数据到一个`pandas` DataFrame。
2. 利用`rename`函数更改列名,注意这里用到了字典形式的参数,键是原列名,值是新列名。
3. 最后使用`to_csv`函数将更新后的DataFrame保存为CSV文件。
如果你需要的是在每个行的'id'列上生成新的UUID,记得在`rename`之前添加一行生成新的'uuid'列的代码。
阅读全文