如何在Plotly中设置可复用布局？Python绘图布局复用问询

我在小型Python项目中设计Plotly图表时，希望保持风格统一，想把定义好的布局作为对象或变量复用，但尝试了几种方法都没成功。

现有代码：

import plotly.express as px

df = px.data.gapminder().query("country=='Canada'")
fig = px.line(df, x="year", y="lifeExp", title='Life expectancy in Canada')
fig.show()

# 基础布局调整
fig.update_layout(xaxis_range=[1970, 2000])
fig.update_layout(font_size=20)
fig.show()

我尝试过以下两种方式，但均无效：

desired_layout = dict(font_size = 20, xaxis_range=[1970, 2000])

fig = px.line(df, x="year", y="lifeExp", title='Life expectancy in Canada', layout = desired_layout)
# 或者：
fig = px.line(df, x="year", y="lifeExp", title='Life expectancy in Canada', **desired_layout)

可行方案 #

方案1：直接调用update_layout传入布局字典 #

Plotly的update_layout方法支持直接传入字典参数，先定义好布局字典，创建图表后调用该方法即可应用布局：

import plotly.express as px

# 定义复用的布局字典
desired_layout = {
    "font_size": 20,
    "xaxis_range": [1970, 2000]
}

df = px.data.gapminder().query("country=='Canada'")
# 创建图表
fig = px.line(df, x="year", y="lifeExp", title='Life expectancy in Canada')
# 应用布局
fig.update_layout(desired_layout)
fig.show()

# 其他图表复用布局
df_us = px.data.gapminder().query("country=='United States'")
fig_us = px.line(df_us, x="year", y="lifeExp", title='Life expectancy in US')
fig_us.update_layout(desired_layout)
fig_us.show()

方案2：封装成复用函数 #

如果需要频繁创建同风格的图表，可以把创建+布局的逻辑封装成函数，减少重复代码：

import plotly.express as px

def create_styled_line_chart(df, x_col, y_col, title, layout_dict):
    fig = px.line(df, x=x_col, y=y_col, title=title)
    fig.update_layout(layout_dict)
    return fig

# 定义统一布局
desired_layout = {"font_size": 20, "xaxis_range": [1970, 2000]}

# 创建加拿大的图表
df_canada = px.data.gapminder().query("country=='Canada'")
fig_canada = create_styled_line_chart(df_canada, "year", "lifeExp", "Life expectancy in Canada", desired_layout)
fig_canada.show()

# 创建美国的图表
df_us = px.data.gapminder().query("country=='United States'")
fig_us = create_styled_line_chart(df_us, "year", "lifeExp", "Life expectancy in US", desired_layout)
fig_us.show()

方案3：自定义Plotly模板（全局风格统一） #

如果项目中所有图表都需要统一风格，推荐定义自定义模板，创建图表时直接指定模板即可：

import plotly.express as px
import plotly.io as pio

# 基于默认模板复制并修改
custom_template = pio.templates["plotly"].copy()
custom_template.layout.font.size = 20
custom_template.layout.xaxis.range = [1970, 2000]

# 注册自定义模板
pio.templates["custom_style"] = custom_template

# 使用模板创建图表
df = px.data.gapminder().query("country=='Canada'")
fig = px.line(df, x="year", y="lifeExp", title='Life expectancy in Canada', template="custom_style")
fig.show()

# 其他图表直接复用模板
df_us = px.data.gapminder().query("country=='United States'")
fig_us = px.line(df_us, x="year", y="lifeExp", title='Life expectancy in US', template="custom_style")
fig_us.show()

内容的提问来源于stack exchange，提问作者Yanirmr