如何解决Colab中Pandas出现的KeyError: 'RS'问题？

错误原因 #

你碰到的KeyError: 'RS'，是因为你的DataFramedf里没有名为'RS'的列。代码第17行df[state].unique()试图访问列名为'RS'的数据，但你的数据集根本不存在这个列名。

第一步：确认数据集结构 #

先运行以下代码，查看你的DataFrame所有列名，找到存储州信息的正确列名：

print(df.columns.tolist())

或者直接查看数据集前几行，明确数据的存储格式：

print(df.head())

修正核心代码 #

结合你的葡萄牙语注释，推测你的数据里州信息可能存在'estado'列，修改代码如下：

# SELEÇÃO DO ESTADO
state = 'RS'
# 筛选出RS州的所有数据
rs_data = df[df['estado'] == state]

Plotly新冠疫情图表实现示例 #

以下是基于Plotly的两种常见疫情可视化代码，适配你的项目需求：

1. 确诊病例趋势折线图 #

import plotly.express as px

# 假设数据集包含日期列'data'、确诊病例列'casos'
fig = px.line(
    rs_data,
    x='data',
    y='casos',
    title='RS州新冠确诊病例趋势',
    labels={'data': '日期', 'casos': '累计确诊数'},
    template='plotly_white'
)
fig.show()

2. 每日新增病例柱状图 #

# 先计算每日新增病例（基于累计数据）
rs_data['novos_casos'] = rs_data['casos'].diff().fillna(0)

fig = px.bar(
    rs_data,
    x='data',
    y='novos_casos',
    title='RS州每日新增新冠确诊病例',
    labels={'data': '日期', 'novos_casos': '新增病例数'},
    color='novos_casos',
    color_continuous_scale='Reds'
)
fig.show()

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