使用 altair 将标识线添加到散点图

Question

我创建了一个基本的散点图来使用 altair 比较两个变量。我希望这些变量具有很强的相关性，并且这些点最终应该位于或接近同一条线。

如何将身份线添加到情节中？

我希望它是一条类似于mark_rule 创建的线，但沿对角线延伸而不是垂直或水平。

这是我所得到的：

import altair as alt
import numpy as np
import pandas as pd

norm = np.random.multivariate_normal([0, 0], [[2, 1.8],[1.8, 2]], 100)

df = pd.DataFrame(norm, columns=['var1', 'var2'])

chart = alt.Chart(df, width=500, height=500).mark_circle(size=100).encode(
    alt.X('var1'),
    alt.Y('var2'),
).interactive()

line = alt.Chart(
    pd.DataFrame({'var1': [-4, 4], 'var2': [-4, 4]})).mark_line().encode(
            alt.X('var1'),
            alt.Y('var2'),
).interactive()

chart + line

这个例子的问题是，当缩放时线条不会永远延伸（就像一个标尺标记），并且绘图会自动缩放到线条的末端，而不仅仅是点。

Answer 1

这并不完美，但您可以将线拉长并设置比例域。

import altair as alt
import numpy as np
import pandas as pd

norm = np.random.multivariate_normal([0, 0], [[2, 1.8],[1.8, 2]], 100)

df = pd.DataFrame(norm, columns=['var1', 'var2'])

chart = alt.Chart(df, width=500, height=500).mark_circle(size=100).encode(
    alt.X('var1', scale=alt.Scale(domain=[-4,4])),
    alt.Y('var2', scale=alt.Scale(domain=[-4,4])),
).interactive()

line = alt.Chart(
    pd.DataFrame({'var1': [-100, 100], 'var2': [-100, 100]})).mark_line().encode(
            alt.X('var1'),
            alt.Y('var2'),
).interactive()

chart + line