Altair大数据集散点图问题求助：RSI与周收益可视化失败

核心问题排查与修复 #

1. 解决data_server导致的空图问题 #

• 先确保Colab环境正确安装并启用数据服务：

  !pip install altair_data_server
  

• 导入并启用data_transformer，同时强制清理数据空值与非数值型数据（这是空图的最常见原因）：

  import altair as alt
  import pandas as pd
  from altair_data_server import data_server
  
  # 加载数据后先做清洗
  df = pd.read_csv("你的数据集路径")
  # 转为数值型，强制错误值为空
  df["RSI"] = pd.to_numeric(df["RSI"], errors="coerce")
  df["1_week_return"] = pd.to_numeric(df["1_week_return"], errors="coerce")
  # 删除关键列空值，保留目标证券
  df = df.dropna(subset=["RSI", "1_week_return"])
  df = df[df["symbol"].isin(["AAPL", "F", "ED"])]
  
  # 启用data_server处理大数据
  alt.data_transformers.enable("data_server")
  

2. 修复图表数据显示异常问题 #

• 手动锁定坐标轴范围（RSI固定0-100，收益率根据你的数据调整，比如-25%到25%），避免极端值压缩数据：
• 缩小散点尺寸并降低透明度，避免大数量点重叠导致视觉上的“数据缺失”：

  # 定义坐标轴
  x_axis = alt.X("RSI:Q", title="相对强弱指数(RSI)", scale=alt.Scale(domain=[0, 100]))
  y_axis = alt.Y("1_week_return:Q", title="1周收益率", scale=alt.Scale(domain=[-0.25, 0.25]))
  

3. 实现证券筛选与缩放平移功能 #

• 添加单选项筛选器，绑定目标证券列表；通过interactive()直接启用缩放平移：

  # 创建证券选择器
  stock_selector = alt.selection_single(
      fields=["symbol"],
      bind=alt.binding_select(options=["AAPL", "F", "ED"], name="选择证券："),
      init={"symbol": "AAPL"}  # 默认选中AAPL
  )
  
  # 绘制完整图表
  chart = alt.Chart(df).mark_circle(size=2, opacity=0.6).encode(
      x=x_axis,
      y=y_axis,
      color=alt.Color("symbol:N", title="证券"),
      tooltip=["symbol", "RSI", "1_week_return"]
  ).add_selection(
      stock_selector
  ).transform_filter(
      stock_selector
  ).interactive()  # 启用缩放平移
  
  chart.show()
  

常见问题排查清单 #

• 空图：检查数据是否存在空值/非数值型值；确认data_server安装后重启Colab内核；检查坐标轴范围是否完全超出数据区间
• 数据显示异常：确认RSI和1_week_return是数值型；调整散点size和opacity避免重叠；检查筛选条件是否误过滤全部数据
• 性能问题：3.8万行属于Altair可处理范围，若仍卡顿可尝试采样数据（df.sample(n=10000)），或改用以下替代方案

替代实现方案（Plotly） #

如果Altair仍有兼容性问题，用Plotly Express处理大数据更流畅，同样支持所有交互需求：

import plotly.express as px

fig = px.scatter(
    df,
    x="RSI",
    y="1_week_return",
    color="symbol",
    range_x=[0, 100],
    range_y=[-0.25, 0.25],
    title="RSI与1周收益率相关性",
    hover_data=["symbol", "RSI", "1_week_return"]
)

# 添加下拉筛选器
fig.update_layout(
    updatemenus=[
        dict(
            buttons=[
                dict(label="全部", method="update", args=[{"visible": [True, True, True]}]),
                dict(label="AAPL", method="update", args=[{"visible": [True, False, False]}]),
                dict(label="F", method="update", args=[{"visible": [False, True, False]}]),
                dict(label="ED", method="update", args=[{"visible": [False, False, True]}])
            ],
            direction="down",
            showactive=True
        )
    ],
    xaxis_title="相对强弱指数(RSI)",
    yaxis_title="1周收益率"
)

fig.show()

内容的提问来源于stack exchange，提问作者Hafemeyer