嗨，我完全懂你遇到的困扰——用apply(wrap)处理长轴标签后，热力图的颜色映射乱了，还出现了空白单元格，连单元格的数值标签也消失了。其实问题出在**wrap函数返回的是列表类型**，而Altair期望轴标签是字符串，同时原始数据的数值绑定也因为数据类型变化失效了，才导致这些异常。

下面给你两种靠谱的解决方案，都能完美解决这个问题：

方法一：在Altair层面直接处理（推荐，不修改原始数据） #

不用动Pandas的原始数据，直接通过Altair的axis.labelExpr用Vega表达式实现标签换行。这种方式不会破坏数据结构，能完美保留颜色映射和单元格数值。

比如我们要把标签按每10个字符换行，代码示例如下：

import altair as alt

# 假设你的数据集是df，包含l_x、l_y（轴标签）和value（热力图数值）列
heatmap = alt.Chart(df).mark_rect().encode(
    # X轴标签处理：每10个字符换行
    x=alt.X(
        'l_x:N',
        title='X轴标题',
        axis=alt.Axis(
            labelExpr="split(datum.label, /(?<=.{10})/).join('\n')"
        )
    ),
    # Y轴标签处理：同样每10个字符换行
    y=alt.Y(
        'l_y:N',
        title='Y轴标题',
        axis=alt.Axis(
            labelExpr="split(datum.label, /(?<=.{10})/).join('\n')"
        )
    ),
    # 保留原来的颜色映射逻辑
    color=alt.Color('value:Q', title='数值大小'),
    # 保留tooltip显示原始标签和数值
    tooltip=['l_x:N', 'l_y:N', 'value:Q']
).properties(
    width=600,
    height=400
)

heatmap.show()

这里的正则表达式(?<=.{10})会匹配每10个字符后的位置，split分割后用'\n'连接，实现自动换行，而且原始数据的所有字段都保持原样，完全不会影响颜色和单元格值的显示。

方法二：修改Pandas数据（返回带换行符的字符串） #

如果你一定要在Pandas里处理标签，那别用默认的wrap函数（它返回列表），自己写一个返回带换行符的字符串的函数：

def wrap_text(text, width=10):
    # 把长文本按指定宽度分割，用换行符连接成字符串
    return '\n'.join([text[i:i+width] for i in range(0, len(text), width)])

# 处理轴标签列
df['l_x'] = df['l_x'].apply(wrap_text, args=[10])
df['l_y'] = df['l_y'].apply(wrap_text, args=[10])

# 正常绘制热力图即可
heatmap = alt.Chart(df).mark_rect().encode(
    x=alt.X('l_x:N', title='X轴标题'),
    y=alt.Y('l_y:N', title='Y轴标题'),
    color=alt.Color('value:Q', title='数值大小'),
    tooltip=['l_x:N', 'l_y:N', 'value:Q']
).properties(width=600, height=400)

heatmap.show()

这种方式处理后，l_x和l_y还是字符串类型，只是包含了换行符，Altair能正确识别，自然不会破坏原来的可视化效果。

总结 #

核心问题就是不要让轴标签变成列表类型，要么在Altair层面用表达式处理，要么在Pandas里返回带换行符的字符串，这样既能实现长标签换行，又能完全保留热力图的颜色逻辑和单元格数值显示。

备注：内容来源于stack exchange，提问作者aborruso