如何通过 Altair 中的变量为散点图着色？

Altair 是 Python 中一款强大的可视化库，它支持通过变量为散点图着色。本文将介绍如何通过 Altair 中的变量为散点图着色。

步骤

1. 导入所需模块

import altair as alt
from vega_datasets import data

2. 加载数据集

source = data.cars()

3. 创建散点图

chart = alt.Chart(source).mark_circle().encode(
    x='Horsepower',
    y='Miles_per_Gallon'
)

4. 为散点图着色

chart = chart.encode(
    color='Origin:N'
)

这里的 N 表示颜色变量是标称的（Nominal），即颜色对应的分类是无序的。

5. 展示图形

chart.show()

完整代码片段如下：

import altair as alt
from vega_datasets import data

# 加载数据集
source = data.cars()

# 创建散点图
chart = alt.Chart(source).mark_circle().encode(
    x='Horsepower',
    y='Miles_per_Gallon'
)

# 为散点图着色
chart = chart.encode(
    color='Origin:N'
)

# 展示图形
chart.show()

结论

通过在 Altair 中使用变量为散点图着色，可以更好地展示数据集中的分类信息，使图形更加易读易懂。