Python 将R值（相关性）添加到Altair中的散点图中

因此，我正在使用Cars数据集，并希望将R值添加到散点图中。因此，我可以使用此代码生成散点图，使用

transform\u regression

添加回归线，这非常好

from vega_datasets import data
import altair as alt
import pandas as pd
import numpy as np

cars = data.cars()
chart = alt.Chart(cars).mark_circle().encode(
        alt.X('Miles_per_Gallon', scale=alt.Scale(domain=(5,50))),
        y='Weight_in_lbs'
)

chart + chart.transform_regression('Miles_per_Gallon','Weight_in_lbs').mark_line()

这是图表

然后我想得到R值。因此，我可以将pandas与此代码一起使用，因为我不确定如何使用Altair获得R值

corl = cars[['Miles_per_Gallon','Weight_in_lbs']].corr().iloc[0,1]
corl

---

现在我想知道如何在图表上添加R值作为一种标签

您可以通过添加文本层来完成此操作：

text = alt.Chart({'values':[{}]}).mark_text(
    align="left", baseline="top"
).encode(
    x=alt.value(5),  # pixels from left
    y=alt.value(5),  # pixels from top
    text=alt.value(f"r: {corl:.3f}"),
)

chart + text + chart.transform_regression('Miles_per_Gallon','Weight_in_lbs').mark_line()

---

在未来版本的Altair中，图表中的空数据将不再需要。

感谢您的帮助。是的，df解决方案不是最直观的。我期待您的数据解决方案。