2744557306 发表于 2024-3-21 09:59

使用python脚本来解决 .ipynb 文件转换为 .py

import json

def convert_ipynb_to_py(ipynb_file, py_file):
    with open(ipynb_file, 'r',encoding='utf-8') as f:
        notebook = json.load(f)

    with open(py_file, 'w',encoding='utf-8') as f:
        for cell in notebook['cells']:
            if cell['cell_type'] == 'code':
                f.write(''.join(cell['source']) + '\n\n')
以上代码是一个用于将.ipynb格式的Jupyter Notebook文件转换为.py格式的Python脚本的函数。该函数会读取指定的.ipynb文件,提取其中的代码单元格内容,并将其写入到指定的.py文件中。
函数的参数解释如下:

1.ipynb_file:输入的.ipynb文件路径。
2.py_file:输出的.py文件路径。

函数的实现步骤:

3.使用json模块读取.ipynb文件,将其解析为一个Python对象。
4.使用open()函数以写入模式打开指定的.py文件。
5.遍历.ipynb文件中的每个单元格。
6.如果单元格类型是代码单元格('code'),将其源代码内容写入到.py文件中。
7.最终关闭文件。

要使用这个脚本,你只需要调用 convert_ipynb_to_py 函数,并传入 .ipynb 文件和期望的 .py 文件名:convert_ipynb_to_py('your_notebook.ipynb', 'your_notebook.py')将 ‘your_notebook.ipynb’ 和 ‘your_notebook.py’ 替换为你的文件名。




页: [1]
查看完整版本: 使用python脚本来解决 .ipynb 文件转换为 .py