Python Pandas DataFrame 数据导出到 Excel 文件

使用 Pandas 将预测结果保存到 Excel

在使用机器学习模型进行预测后，您可能需要将结果保存到 Excel 文件中，以便于后续分析或分享。

假设您已经使用 model.predict() 函数获得了预测结果 pre，并希望将其保存到名为 'result.xlsx' 的 Excel 文件中。

以下是使用 Pandas 库实现此目标的 Python 代码示例：

import pandas as pd

# 假设 pre 是您的预测结果
pre = model.predict(daice) 

# 创建 Pandas DataFrame
result_df = pd.DataFrame(pre, columns=['厚度', '孔隙率', '压缩回弹性率'])

# 将 DataFrame 保存到 Excel 文件
result_df.to_excel('result.xlsx', index=False)

代码解释：

1. import pandas as pd: 导入 Pandas 库并将其简称为 pd。
2. result_df = pd.DataFrame(pre, columns=['厚度', '孔隙率', '压缩回弹性率']):
   • 使用预测结果 pre 和指定的列名创建 Pandas DataFrame。
   • DataFrame 是一个二维数据结构，类似于 Excel 表格。
3. result_df.to_excel('result.xlsx', index=False):
   • 调用 to_excel() 函数将 DataFrame 保存到名为 'result.xlsx' 的 Excel 文件中。
   • index=False 参数确保不将 DataFrame 的索引列写入 Excel 文件。

通过以上步骤，您就可以轻松地将机器学习模型的预测结果保存到 Excel 文件中了。