数学建模社区-数学中国

标题: 使用python脚本来解决 .ipynb 文件转换为 .py [打印本页]
作者: 2744557306    时间: 2024-3-21 09:59
标题: 使用python脚本来解决 .ipynb 文件转换为 .py
  1. import json
    4 ^( @! Z% G2 n& V( e% E# B. O6 h2 d

  2. . i" k$ I) _9 F
  3. def convert_ipynb_to_py(ipynb_file, py_file):
    7 }- Y- d! O; g" j6 \. l+ J3 g
  4.     with open(ipynb_file, 'r',encoding='utf-8') as f:
    6 Y  C. J, q% H( ]4 {1 I9 \9 ]
  5.         notebook = json.load(f)
    2 g" E  t% O1 m
  6. ( O3 C" ~% o& ?. o6 g7 l
  7.     with open(py_file, 'w',encoding='utf-8') as f:
    7 e+ n7 o' I9 ?% |4 z
  8.         for cell in notebook['cells']:
    : i, U- ]) o1 }2 f
  9.             if cell['cell_type'] == 'code':
    : y4 n% d; S' C1 E5 b% P) e$ d' |
  10.                 f.write(''.join(cell['source']) + '\n\n')
    & Q9 B* p7 |% a! r; H+ G3 Y
复制代码
以上代码是一个用于将.ipynb格式的Jupyter Notebook文件转换为.py格式的Python脚本的函数。该函数会读取指定的.ipynb文件，提取其中的代码单元格内容，并将其写入到指定的.py文件中。
+ _- j* M+ i7 K& C' O4 M6 ?函数的参数解释如下：
- k  g; B, [4 o, J. C; ^2 m8 _1 y$ |  ]5 ]- K) D* e
1.ipynb_file：输入的.ipynb文件路径。
* t* A# U: m+ B* U: U7 K" B: z2.py_file：输出的.py文件路径。  F6 k+ O9 z, x; s! R9 }. g9 h

0 @( Q! i- u! G# y  V$ F函数的实现步骤：, P( j. S% O0 S+ I& D* B
( [7 f9 y4 {8 X% k. D
3.使用json模块读取.ipynb文件，将其解析为一个Python对象。% Q: y! @6 l6 a8 P  u
4.使用open()函数以写入模式打开指定的.py文件。
8 ]: l0 j3 B% x" B( `2 W5.遍历.ipynb文件中的每个单元格。
' y8 Z: k+ M6 g: ]) a7 @6.如果单元格类型是代码单元格（'code'），将其源代码内容写入到.py文件中。" T# c+ d8 g) E: J( T
7.最终关闭文件。
% _4 m! ~4 ?6 R: ^8 ]& ^1 ]/ O! h! j+ k/ A( }
要使用这个脚本，你只需要调用 convert_ipynb_to_py 函数，并传入 .ipynb 文件和期望的 .py 文件名：
  1. convert_ipynb_to_py('your_notebook.ipynb', 'your_notebook.py')
复制代码
将 ‘your_notebook.ipynb’ 和 ‘your_notebook.py’ 替换为你的文件名。
/ P: _4 [5 C$ R
1 ]9 S  h+ @; d  W- f; e3 ~3 r/ V" e1 J% p. W1 i& K

6 {  x+ S+ Q/ c+ @5 |) _8 M8 R% [$ w5 n& m9 x+ ~




欢迎光临 数学建模社区-数学中国 (http://www.madio.net/) Powered by Discuz! X2.5