如何插入一些"线"从另一个脚本进入主脚本

时间:2018-04-03 08:55:55

标签: python import refactoring

假设我有这个脚本:

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'对象

重用这些线路的正确方法是什么?

1 个答案:

答案 0 :(得分:2)

使用功能

# 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")