jupyter notebook对输出结果的标注

Question

Python输出dataFrame结果到Excel中后，希望有以下操作：

新建txt文件，记录Excel是来自Python（jupyter notebook）的哪一个code文件，以及对输出数据的描述或注解，从而方便之后查看

请问什么代码可以做到？

Answer 1

引自免费微信小程序：皆我百晓生

要实现这个需求，你需要在Jupyter Notebook中运行Python代码时，同时创建一个txt文件，并将相关信息写入其中。下面是一个示例代码，展示了如何完成这项任务。

首先，在你的Python代码开始部分，定义一些变量来保存信息，例如当前代码文件名、注解内容等。然后，使用pandas库处理DataFrame并将其写入Excel，最后将这些信息写入txt文件。

import os
import pandas as pd

# 保存相关信息的变量
current_file = os.path.abspath(__file__)  # 获取当前Python文件的绝对路径
notebook_name = os.path.splitext(os.path.basename(current_file))[0]  # 提取文件名（去除扩展名）
description = "这是我对输出数据的描述或注解"  # 描述或注解内容

# 假设有一个名为df的数据框，我们将其写入Excel
output_excel_file = 'output.xlsx'
df.to_excel(output_excel_file, index=False)  # 将DataFrame写入Excel，不包含索引

# 创建txt文件并写入相关信息
with open(f'{notebook_name}_info.txt', 'w') as f:
    f.write(f"原始代码文件：{current_file}\n")
    f.write(f"输出Excel文件：{output_excel_file}\n")
    f.write(f"描述或注解：\n{description}\n")

这段代码会执行以下操作：

1. 获取当前正在运行的Python文件的绝对路径。
2. 提取文件名（去除.py或.ipynb等扩展名）。
3. 定义描述或注解内容。
4. 将DataFrame写入名为output.xlsx的Excel文件。
5. 在同一个目录下创建一个名为<notebook_name>_info.txt的文本文件，将原始代码文件路径、输出Excel文件路径和描述或注解内容写入其中。

确保在运行此代码之前已经安装了pandas库，如果没有安装，可以通过pip安装：

pip install pandas

这样，每次运行你的Python代码时，都会自动创建一个txt文件，记录了相关的信息，便于后续查看。