循环中动态切换变量问题：Plotly绘图无法迭代调用group_n变量

原代码核心问题 #

1. 变量名错误：代码中froup是未定义的拼写错误，对应长度的分组变量命名混乱；且手动命名group_1/group_2/group_3的方式，无法通过group_ + i这种字符串拼接方式调用，Python不支持直接通过动态字符串获取变量。
2. 分组与物种顺序不匹配：species列表顺序是["Adelie", "Gentoo", "Chinstrap"]，但手动定义的group_2对应Chinstrap、group_3对应Gentoo，循环时索引对应会出错。

优化方案 #

放弃手动命名单个分组变量，改用列表/字典统一存储分组数据，让循环可以直接按索引或键取值，避免动态变量名的问题。

方法一：用列表存储分组（对应species顺序）

import plotly.graph_objects as go
fig = go.Figure()

# 调整物种顺序，和分组数据一一对应
species = ["Adelie", "Chinstrap", "Gentoo"]  

# 用列表存储depth和length的分组数据，顺序与species一致
depth_groups = [
    df[df['species'] == 'Adelie']['culmen_depth_mm'],
    df[df['species'] == 'Chinstrap']['culmen_depth_mm'],
    df[df['species'] == 'Gentoo']['culmen_depth_mm']
]

length_groups = [
    df[df['species'] == 'Adelie']['culmen_length_mm'],
    df[df['species'] == 'Chinstrap']['culmen_length_mm'],
    df[df['species'] == 'Gentoo']['culmen_length_mm']
]

# 循环添加trace，通过索引匹配对应分组
for i, sp in enumerate(species):
    fig.add_trace(go.Scatter(
        x=depth_groups[i], 
        y=length_groups[i], 
        mode='markers', 
        name=sp
    ))

fig.update_layout(title='企鹅喙部深度vs长度')
fig.show()

方法二：用字典存储分组（更直观，无需关注顺序）

import plotly.graph_objects as go
fig = go.Figure()

species = ["Adelie", "Gentoo", "Chinstrap"]

# 按物种名作为键，存储对应的depth和length数据
group_data = {}
for sp in species:
    group_data[sp] = {
        'depth': df[df['species'] == sp]['culmen_depth_mm'],
        'length': df[df['species'] == sp]['culmen_length_mm']
    }

# 循环每个物种，直接从字典中取对应数据
for sp in species:
    fig.add_trace(go.Scatter(
        x=group_data[sp]['depth'], 
        y=group_data[sp]['length'], 
        mode='markers', 
        name=sp
    ))

fig.update_layout(title='企鹅喙部深度vs长度')
fig.show()

额外优化建议 #

如果DataFrame结构规范，直接用groupby简化分组代码，避免重复写筛选逻辑：

import plotly.graph_objects as go
fig = go.Figure()

# 按species分组，直接遍历分组对象
for sp, group_df in df.groupby('species'):
    fig.add_trace(go.Scatter(
        x=group_df['culmen_depth_mm'], 
        y=group_df['culmen_length_mm'], 
        mode='markers', 
        name=sp
    ))

fig.update_layout(title='企鹅喙部深度vs长度')
fig.show()

这种方式无需手动定义任何分组变量，代码更简洁且不易出错。

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