Altair：提取和顯示回歸系數

Question

這個問題解決了如何使用mark_text()訪問和顯示 R ²值

我有興趣訪問和顯示系數。 如文檔中所述，將rSquared替換為coef會產生截距和斜率的扁平數組。

如何索引到這個數組以僅顯示其中一個值，例如斜率？ 我想知道mark_text()步驟之前是否應該進行轉換（可能是transform_filter() ，或者是否可以使用altair.Text() 。

我知道其他方法涉及單獨確定此信息，然后將其添加為附加層。

抱歉，如果這是一個非常簡單的問題。 提前致謝。

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

np.random.seed(42)
x = np.linspace(0, 10)
y = x - 5 + np.random.randn(len(x))

df = pd.DataFrame({'x': x, 'y': y})

chart = alt.Chart(df).mark_point().encode(
    x='x',
    y='y'
)
line = chart.transform_regression('x', 'y').mark_line()

params = alt.Chart(df).transform_regression(
    'x', 'y', params=True
).mark_text(align='left').encode(
    x=alt.value(20),  # pixels from left
    y=alt.value(20),  # pixels from top
    text='rSquared:N',

    # text='coef:N' # flattened array
    # text='coef[0]:N' # fails
)

chart + line + params

Answer 1

您可以使用計算轉換來訪問它：

params = alt.Chart(df).transform_regression(
    'x', 'y', params=True
).transform_calculate(
    intercept='datum.coef[0]',
    slope='datum.coef[1]',
).mark_text(align='left').encode(
    x=alt.value(20),  # pixels from left
    y=alt.value(20),  # pixels from top
    text='intercept:N'
)

chart + line + params