如何将其他脚本中的一些“行”插入到主脚本中

Question

假设我有这个脚本：

from bokeh.plotting import figure, show, output_file
p = figure()
p.circle([1,2,3], [4,5,6])
p.title.text = "Title"
p.title.text_color = "Orange"
p.title.text_font = "times"
show(p)
output_file("file.html")

我希望在其他脚本中重用第4行到第6行，而不必在每个脚本中复制和粘贴它们。

如果我将第4-6行放在一个单独的.py文件中，然后将该文件导入到主脚本中，那么将会有一个关于未定义的'p‘对象的NameError。

重用这些行的正确方法是什么？

Answer 1

使用函数

# in settitle.py
def set_title(p):
    p.title.text = "Title"
    p.title.text_color = "Orange"
    p.title.text_font = "times"

并像这样导入函数

from settitle import set_title

并使用

from settitle import set_title
from bokeh.plotting import figure, show, output_file
p = figure()
p.circle([1,2,3], [4,5,6])
set_title(p)
show(p)
output_file("file.html")