使用 altair 在 jupyterlab 中注册自定义 vega 格式化程序

Question

我正在尝试实现一个可以在 jupyterlab 中与 altair 一起使用的自定义格式化程序formatType 。 我不知道如何在 jupyterlab 中注册 vega 表达式 function。 它在使用 vega-editor 并使用 JS 控制台注入表达式时起作用。 如何从 jupyterlab 单元中注册表达式 function？

使用生成的定义

import altair as alt
import pandas as pd

def pub_theme():
    return {'config': {"customFormatTypes": True}}
alt.themes.register('pub', pub_theme)
alt.themes.enable('pub')

alt.Chart(
    pd.DataFrame({'x': [1e-3, 1e-2, 1, 1e10], 'y':[1e-3, 1e-2, 1, 1e10]})
).mark_line(
).encode(
    y=alt.Y('y:Q'),
    x=alt.X('x:Q', axis=alt.Axis(formatType='pubformat'))
)

并在 Vega-lite 编辑器中打开定义按预期给出[Error] Unrecognized function: pubformat

如果pubformat是通过 JS 控制台 witg 定义的，例如VEGA_DEBUG.vega.expressionFunction('pubformat', function() {return 'test'}) ，那么编辑器会生成预期的 output，在 x 轴上有 6 个相同的标签test 。

在 jupyterlab 中实现这一目标的正确方法是什么？

从 jupyterlab 页面上的 JS 控制台注入表达式vega.expressionFunction('pubformat', function() {return 'test'})不起作用。 通过定义它

from IPython.display import HTML
    HTML("""
        <script>
            vega.expressionFunction('pubformat', function() {return 'test'})
        </script>
     """)

也没有用。

如果注入 javascript 是一个问题，是否还有其他选项可以在 altair 中实现自定义格式化程序？ 似乎可以在轴上使用labelExpr实现类似的行为，但是这不太通用，因为必须为每个轴重复表达式。 谢谢！

Answer 1

Vega-Lite 中的自定义格式化程序是在4.11版中添加的； Altair 当前支持Vega-Lite v4.8.1 ，因此当前版本的 Altair 不支持自定义格式化程序。

Answer 2

一种可能的解决方法是使用labelExpr 。 我可以使用

import altair as alt
import pandas as pd

def pub_theme():
    return {'config': {"axis": {"labelExpr": "replace(datum.label, datum.label, 'test')"}}}
alt.themes.register('pub', pub_theme)
alt.themes.enable('pub')

alt.Chart(
    pd.DataFrame({'x': [1e-3, 1e-2, 1, 1e10], 'y':[1e-3, 1e-2, 1, 1e10]})
).mark_line(
).encode(
    y=alt.Y('y:Q'),
    x=alt.X('x:Q')
)