保存无空白的Altair图表

import altair as alt
from PIL import Image
from io import BytesIO

chart = alt.Chart(data).mark_bar().encode(
    x='category',
    y='count'
)

# 将图表保存为JSON格式的Vega-Lite规范
spec = chart.to_json()

# 创建一个空白的PIL图像对象
image = Image.new('RGB', (1, 1))

# 使用Selenium将Vega-Lite规范渲染为图像
image = alt.renderers.screenshot(driver='firefox').encode(spec).to_image()

# 将图像保存到内存中
buffer = BytesIO()
image.save(buffer, format='PNG')
buffer.seek(0)

# 从内存中读取图像
image = Image.open(buffer)

# 去除图像周围的空白边距
image = image.crop(image.getbbox())

# 可选：调整图像大小
# image = image.resize((width, height))

# 保存图像
image.save('chart.png')