如何在Plotly.py中导出正方形宽高比的静态2D散点图？

我想用plotly.py创建2D散点图，并导出为坐标轴在屏幕单位下呈正方形的静态.svg文件（类似Matplotlib中让坐标轴长度等比例的需求，我的x和y数据尺度不同）。

我尝试直接设置图表宽高相等，在只有单个trace且无图例时效果还行：

import numpy as np
import plotly.graph_objects as go

np.random.seed(42)

fig = go.Figure()

fig.add_trace(go.Scatter(
    x=np.random.uniform(0, 1, 50), 
    y=np.random.randint(1, 100, 50), 
    mode='markers',
    ))

fig.update_layout(width=500, height=500)

# 注：Plotly生成静态图片需要安装Kaleido（pip install -U kaleido）
fig.write_image("example.svg")

输出效果：单个trace、无图例时，坐标轴呈现正方形比例

但添加多个trace和图例后，图形就变形了：

import numpy as np
import plotly.graph_objects as go

np.random.seed(42)

fig = go.Figure()

fig.add_trace(go.Scatter(
    x=np.random.uniform(0, 1, 50), 
    y=np.random.randint(1, 100, 50), 
    mode='markers',
    name="SomeLongTraceLabel1"
    ))

fig.add_trace(go.Scatter(
    x=np.random.uniform(0, 1, 50), 
    y=np.random.randint(1, 100, 50), 
    mode='markers',
    name="SomeLongTraceLabel2"
    ))

fig.update_layout(width=500, height=500)

# 注：Plotly生成静态图片需要安装Kaleido（pip install -U kaleido）
fig.write_image("example.svg")

输出效果：带图例的情况下，坐标轴不再是正方形比例

我知道3D图可以用layout.scene.aspectmode="cubic"实现类似效果，但没找到2D图的对应设置。有没有办法不用每次根据图例长度调整宽高，就能自动生成坐标轴为正方形比例的图形？

Plotly 2D图可以通过设置坐标轴的比例锚定和缩放比例来实现屏幕单位下的正方形坐标轴，同时配合布局调整避免图例干扰，具体步骤如下：

1. 强制坐标轴比例为正方形
   使用xaxis.scaleanchor和yaxis.scaleratio参数，将x轴锚定到y轴，设置缩放比例为1，这样无论数据尺度如何，坐标轴在屏幕上的长度都会保持相等：

fig.update_xaxes(scaleanchor="y", scaleratio=1)

2. 固定图例位置并调整边距
   为了避免图例占用空间导致绘图区域变形，可以将图例固定到绘图区域外部（比如右侧），并设置合适的边距，确保绘图区域保持正方形。

完整示例代码 #

import numpy as np
import plotly.graph_objects as go

np.random.seed(42)

fig = go.Figure()

fig.add_trace(go.Scatter(
    x=np.random.uniform(0, 1, 50), 
    y=np.random.randint(1, 100, 50), 
    mode='markers',
    name="SomeLongTraceLabel1"
))

fig.add_trace(go.Scatter(
    x=np.random.uniform(0, 1, 50), 
    y=np.random.randint(1, 100, 50), 
    mode='markers',
    name="SomeLongTraceLabel2"
))

# 核心设置：强制坐标轴比例为正方形
fig.update_xaxes(scaleanchor="y", scaleratio=1)

# 固定图例位置并调整布局边距，确保绘图区域不受图例影响
fig.update_layout(
    width=600,  # 预留图例空间，总宽度大于高度
    height=500,
    legend=dict(
        x=1.05,  # 图例放在绘图区域右侧外部
        y=0.5,
        xanchor="left"
    ),
    margin=dict(
        l=50,
        r=150,  # 右侧预留足够空间放图例
        b=50,
        t=50,
        pad=4
    )
)

# 导出为SVG
fig.write_image("square_axis_example.svg")

关键说明 #

• scaleanchor="y"表示x轴的缩放锚定到y轴，scaleratio=1确保x轴和y轴的屏幕单位比例为1:1，无论数据范围差异多大，坐标轴在视觉上都是正方形。
• 通过调整总宽度和右侧边距，为图例预留固定空间，避免图例挤压绘图区域导致比例变形。
• 如果不需要图例在外部，也可以将图例放在绘图区域内部（比如右上角），此时无需额外调整宽高，只要设置scaleanchor和scaleratio即可保证坐标轴比例。

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