ECG数据框多列同时切片报错修复及切片命名方法咨询

问题分析 #

你遇到的list assignment index out of range错误，核心原因有三个：

1. 空列表cut直接通过索引cut[position]赋值，列表不存在对应位置的元素，触发索引越界
2. 循环逻辑混乱：外层遍历导联、内层遍历峰值位置，会导致同一导联被重复赋值给不同索引
3. 参数名使用了Python内置函数len，容易引发命名冲突

同时原代码未实现cut[1]、cut[2]…的命名需求，需要调整存储结构。

修复后的代码（两种场景） #

场景1：每个峰值对应所有导联的切片（cut[1]对应第一个峰值的12导联数据） #

如果pos列表包含多个峰值位置，需为每个峰值生成一组全导联切片：

def slicing(patient, pos, window_len):
    # 用字典存储切片，键从1开始计数，对应cut[1]、cut[2]…
    cut = {}
    # 遍历每个峰值的顺序索引和位置
    for idx, position in enumerate(pos, start=1):
        # 计算切片起止位置，防止起始位置小于0
        start = max(0, position - window_len)
        end = position + window_len
        # 一次性截取该峰值前后所有导联的数据
        cut[idx] = patient.loc[start:end, :]
    return cut

场景2：单峰值下每个导联单独切片（cut[1]对应第一个导联的切片） #

如果pos仅包含第一列的最高峰值（符合你“以第一列最高峰值为基准”的需求），需为每个导联生成单独切片：

def slicing(patient, pos, window_len):
    cut = {}
    # 取第一列的最高峰值位置
    peak_pos = pos[0]
    start = max(0, peak_pos - window_len)
    end = peak_pos + window_len
    # 遍历每个导联，按1、2…命名切片
    for idx, lead in enumerate(patient.columns, start=1):
        cut[idx] = patient.loc[start:end, lead]
    return cut

关键优化点 #

• 用字典替代列表：避免空列表索引越界问题，同时支持cut[1]这种命名方式
• 处理边界异常：用max(0, ...)确保切片起始位置不会小于0，防止数据越界
• 避免命名冲突：把参数len改为window_len，不再覆盖Python内置函数
• 简化切片逻辑：直接对数据框整体/列切片，比逐个遍历导联更高效

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