Pandas向DataFrame列首添加字符串与列表报错求助

问题场景 #

要在生成的Excel表格中，给metricID列开头添加字符串"timeframe"、consumo列开头添加字符串"2022-11-10 2022-12-10"，同时要求新增行在表格最开头，后续原有列表元素按consumo降序排序。直接拼接字符串与列表时触发TypeError。

原工作代码 #

def CreateExcel(metric,consumo, writer):
    df = pd.DataFrame({
        'metricID': metric,
        'consumo' : consumo,
    })
    df = df.sort_values('consumo', ascending=False)
    df.to_excel(writer, sheet_name="foglio1", startrow=1, header=False, index=False)
    workbook = writer.book
    worksheet = writer.sheets["foglio1"]
    (max_row, max_col) = df.shape
    column_settings = [{"header":column} for column in df.columns]
    worksheet.add_table(0, 0, max_row, max_col - 1, {"columns":column_settings})
    worksheet.set_column(0, max_col - 1, 70)

错误尝试与报错信息 #

尝试直接拼接字符串和列表：

df = pd.DataFrame({
        'metricID': "timeframe" + metric,
        'consumo' : "2022-11-10 2022-12-10" + consumo,
    })

触发报错：

'metricID': "timeframe" + metric,
            ~~~~~~~~~~~~^~~~~~~
TypeError: can only concatenate str (not "list") to str

解决方案 #

错误原因 #

Python不支持单个字符串与整个列表直接拼接，需要将新增内容作为单独的数据行，与原有数据合并后再处理。

修改后的完整代码 #

def CreateExcel(metric, consumo, writer):
    # 处理原有数据：生成DataFrame并按consumo降序排序
    df_data = pd.DataFrame({
        'metricID': metric,
        'consumo': consumo,
    }).sort_values('consumo', ascending=False)
    
    # 创建需要添加在开头的指定行数据
    df_header = pd.DataFrame({
        'metricID': ["timeframe"],
        'consumo': ["2022-11-10 2022-12-10"]
    })
    
    # 合并两个DataFrame，确保开头行在最顶部
    df = pd.concat([df_header, df_data], ignore_index=True)
    
    # 后续Excel写入与表格设置逻辑保持不变
    df.to_excel(writer, sheet_name="foglio1", startrow=1, header=False, index=False)
    workbook = writer.book
    worksheet = writer.sheets["foglio1"]
    (max_row, max_col) = df.shape
    column_settings = [{"header": column} for column in df.columns]
    worksheet.add_table(0, 0, max_row, max_col - 1, {"columns": column_settings})
    worksheet.set_column(0, max_col - 1, 70)

关键说明 #

1. 先排序再合并：先对原有数据按consumo降序排序，保证后续元素的顺序符合要求
2. 单独处理开头行：将需要添加的指定字符串做成单独的一行DataFrame，避免字符串与列表拼接的错误
3. 合并数据：使用pd.concat将开头行DataFrame放在原有数据上方，ignore_index=True重置索引，避免索引冲突

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