如何子类化Altair的Chart类？解决__init__方法参数报错问题

问题原因 #

Altair的alt.Chart继承自不可变的SchemaBase类，其__init__仅接受与图表Schema匹配的参数（如data、mark、encoding等）。直接在子类__init__中传入align这类属于mark配置的参数，会被父类判定为无效参数，从而抛出TypeError。而通过**kwargs传递参数时，这些配置会被正确解析到对应的mark或配置项中，所以能正常运行。

可行实现方式 #

方式1：在__init__中封装mark配置

将固定的视觉参数（如orient、align）整合到mark参数里，传递给父类：

import altair as alt
from vega_datasets import data

class HBarChart(alt.Chart):
    def __init__(self, data=None, **kwargs):
        # 预设横向柱状图的基础配置
        base_mark = alt.Bar(orient='horizontal', align='center')
        # 合并用户传入的mark配置（如果有）
        if 'mark' in kwargs:
            if isinstance(kwargs['mark'], alt.MarkConfig):
                base_mark = base_mark.configure(**kwargs.pop('mark').to_dict())
        super().__init__(data=data, mark=base_mark, **kwargs)

# 使用示例
source = data.barley()
chart = HBarChart(source).encode(
    x='sum(yield):Q',
    y='variety:N'
)
chart.show()

方式2：用类方法封装实例创建逻辑

这种方式更贴合Altair的不可变设计，避免直接重写__init__的潜在问题：

import altair as alt
from vega_datasets import data

class HBarChart(alt.Chart):
    @classmethod
    def create(cls, data=None, **kwargs):
        # 预设mark配置，允许用户覆盖
        mark_config = {'type': 'bar', 'orient': 'horizontal', 'align': 'center'}
        mark_config.update(kwargs.pop('mark', {}))
        return cls(data=data, mark=mark_config, **kwargs)

# 使用示例
source = data.barley()
chart = HBarChart.create(source).encode(
    x='sum(yield):Q',
    y='variety:N'
)
chart.show()

方式3：极简版（固定mark参数）

如果不需要用户覆盖mark基础配置，可直接传入预设的alt.Bar实例：

import altair as alt
from vega_datasets import data

class HBarChart(alt.Chart):
    def __init__(self, data=None, **kwargs):
        super().__init__(data=data, mark=alt.Bar(orient='horizontal', align='center'), **kwargs)

# 使用示例
source = data.barley()
chart = HBarChart(source).encode(
    x='sum(yield):Q',
    y='variety:N'
)
chart.show()

核心要点 #

所有自定义的视觉参数（如align、orient）必须封装到mark或其他合法的Schema参数中，不能直接作为__init__的独立参数传递给父类，否则会触发参数不匹配的错误。

内容的提问来源于stack exchange，提问作者Víctor López Fandiño