我经常用 Python 编写绘图脚本,我经常想对图形使用通用设置,包括线宽和颜色等。
目前我将所有这些变量复制并粘贴到文件的开头,例如:
import matplotlib.pyplot as plt
company_color_1 = "#40A7E8"
company_color_2 = "#707070"
marker_width = 2.5
# my script here:
plt.plot(x, y, color=company_color_1)
我想要实现的是,我可以在每个脚本的开头放置相同的一行,而不是:
import matplotlib.pyplot as plt
my_graph_settings()
# script:
plt.plot(x, y, color=company_color_1)
这样我就可以通过一个文件控制所有内容并更好地管理更改。
我将如何在 Python 中解决这个问题?
请您参考如下方法:
使用您想要的变量创建一个名为 my_graph_settings.py 的 Python 文件,然后将其导入。如果您不介意潜在的命名冲突,并且希望像在当前脚本中定义的那样使用每个变量,您可以将所有内容导入当前命名空间:
import matplotlib.pyplot as plt
from my_graph_settings import *
# script:
plt.plot(x, y, color=company_color_1)
但是最好给它加上别名,这样你就不会污染命名空间:
import matplotlib.pyplot as plt
import my_graph_settings as mgs
# script:
plt.plot(x, y, color=mgs.company_color_1)
