Python Plotly 散點圖 plot 中的水平線

Question

我正在尋找一種在 Plotly Scatter plot 中繪制兩條水平線的方法。 我的 x 軸索引不是固定的，並且每次都在變化。 所以我在 y = 5 和 y = 18 處尋找一條水平線，水平穿過圖表

我在這里尋找解決方案，但我不確定如何使用 Plotly express 的布局

我的分散代碼 plot：

import plotly.express as px
df = pd.DataFrame({"x":[0, 1, 2, 3, 4,6,8,10,12,15,18], "y":[0, 1, 4, 9, 16,13,14,18,19,5,12]})
fig = px.scatter(df, x="x", y="y")
fig

Answer 1

是的，您可以使用fig.update_layout()來做到這一點，方法如下：

import pandas as pd
import plotly.express as px

df = pd.DataFrame({ "x":[0, 1, 2, 3, 4,6,8,10,12,15,18],
                    "y":[0, 1, 4, 9, 16,13,14,18,19,5,12]})
fig = px.scatter(df, x="x", y="y")

# add two horizontal lines
fig.update_layout(shapes=[
    # adds line at y=5
    dict(
      type= 'line',
      xref= 'paper', x0= 0, x1= 1,
      yref= 'y', y0= 5, y1= 5,
    ),
    # adds line at y=18
    dict(
      type= 'line',
      xref= 'paper', x0= 0, x1= 1,
      yref= 'y', y0= 18, y1= 18,
    )
])

fig.show()

生成此圖：

我不知道是否有更簡單的方法，但這就是我會使用的

Answer 2

這些天使用 vline 或 hline 更簡單

import plotly.express as px

df = pd.DataFrame({"x":[0, 1, 2, 3, 4,6,8,10,12,15,18], "y":[0, 1, 4, 9, 
16,13,14,18,19,5,12]})

fig = px.scatter(df, x="x", y="y")
fig.add_hline(y=5)
fig.add_hline(y=18)

fig.show()