Pandas序列化CSV后DataFrame异常致finplot绘图失败问题排查

问题原因分析 #

1. IndexError: positional indexers are out-of-bounds #

从CSV加载的DataFrame丢失了原有的索引结构（比如时间索引），或者列的顺序/数量与API返回的不一致。finplot依赖DataFrame的索引完成坐标映射，当索引缺失或格式异常时，会触发索引越界错误。

2. TypeError: '>' not supported between instances of 'str' and 'float' #

CSV保存再读取后，原本的数值列（Open/Close/MA系列等）被解析成了字符串类型。finplot绘图时需要对数值进行比较计算，字符串与float类型无法直接运算，因此触发类型不兼容错误。

解决方法 #

方案1：修正CSV读写的索引与类型处理 #

修改get_time_series_data函数，确保保存CSV时保留索引，读取时自动解析索引和数值类型：

def get_time_series_data(symbol):
    if os.path.exists(symbol_path):
        # 读取时指定索引列、解析日期，并强制转换数值列类型
        df = pd.read_csv(
            symbol_path,
            index_col=0,  # 恢复原DataFrame的索引列
            parse_dates=True,  # 自动将索引解析为datetime类型
            dtype={
                'Open': float,
                'Close': float,
                'High': float,
                'Low': float,
                'MA100': float,
                'MA50': float
            }
        )
        return df
    else:        
        df = call_api(symbol)
        # 保存时保留索引（API返回的DataFrame需带有效索引）
        df.to_csv(symbol_path, index=True)        
        return df

方案2：统一DataFrame结构校验（可选） #

在plot函数开头增加校验逻辑，确保输入的DataFrame符合绘图要求：

def plot(df):
    # 强制转换所有数值列为float类型
    num_cols = ['Open','Close','High','Low','MA100','MA50']
    df[num_cols] = df[num_cols].astype(float)
    
    # 确保索引为datetime类型（如果业务依赖时间轴）
    if not pd.api.types.is_datetime64_any_dtype(df.index):
        df.index = pd.to_datetime(df.index)
    
    ax = fplt.create_plot("MY SYMBOL", rows=1)
    dfn = df[['Open','Close','High','Low']]
    fplt.candlestick_ochl(dfn)
    fplt.plot(df['MA100'], ax=ax, legend='ma-100', color='#927', width=3)
    fplt.plot(df['MA50'], ax=ax, legend='ma-50', color='#188bc2', width=3)
    fplt.show()

关键说明 #

• CSV格式默认不会保存DataFrame的索引类型（如datetime），必须显式指定index=True保存，并在读取时用parse_dates=True恢复。
• 若数值列包含特殊格式（如千分位逗号、空值），CSV读取时会自动转为字符串，因此需要强制指定dtype或二次转换类型。

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