Pandas DataFrame排序与插入字符串行：专业教程

心靈之曲

发布时间：2025-08-21 18:30:01

485人浏览过

来源于php中文网

原创

pandas dataframe排序与插入字符串行：专业教程

本文档旨在解决在Pandas DataFrame中对数值列进行排序，并在排序后的DataFrame顶部插入包含字符串数据的行的问题。我们将提供一种清晰、高效的方法，确保排序的正确性，同时保留DataFrame的整体数据结构。通过本文，你将学会如何灵活地处理混合数据类型的DataFrame，并将其导出为Excel文件。

问题分析

在处理DataFrame时，有时需要在数值列排序后，在DataFrame的开头添加一行字符串作为描述或标题。直接将字符串列表与数值列表合并会导致列的数据类型变为字符串，从而无法正确排序。解决此问题的关键在于先对数值数据进行排序，然后再插入字符串行。

解决方案

以下是一种实现该目标的有效方法：

歌者PPT

歌者PPT，AI 写 PPT 永久免费

下载

创建DataFrame: 首先，使用数值数据创建DataFrame。
排序: 使用sort_values()方法按指定的列对DataFrame进行排序。
创建新的DataFrame行: 创建一个新的DataFrame，包含要插入的字符串数据。
合并DataFrame: 使用pd.concat()将新的DataFrame行与排序后的DataFrame合并。
重置索引: 使用reset_index(drop=True)重置DataFrame的索引。

代码示例

import pandas as pd

def create_excel(metric, consumo, writer):
    # 1. 创建DataFrame
    df = pd.DataFrame({
        'metricID': metric,
        'consumo': consumo,
    })

    # 2. 排序
    df = df.sort_values('consumo', ascending=False)

    # 3. 创建新的DataFrame行
    new_row = pd.DataFrame({'metricID': 'Data frame', 'consumo': 'from:2022-12-01 00:00:00 to:2022-12-14 16:13:00'}, index=[0])

    # 4. 合并DataFrame
    df = pd.concat([new_row, df.loc[:]]).reset_index(drop=True)

    # 5. 导出到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)


# 示例数据
metric = ['A', 'B', 'C', 'D']
consumo = [10.5, 5.2, 8.9, 12.1]

# 创建ExcelWriter对象
writer = pd.ExcelWriter('output.xlsx', engine='xlsxwriter')

# 调用函数
create_excel(metric, consumo, writer)

# 保存Excel文件
writer.close()

代码解释

pd.DataFrame(): 创建Pandas DataFrame。
sort_values('consumo', ascending=False): 按'consumo'列降序排序。
pd.DataFrame({'metricID': 'Data frame', 'consumo': 'from:2022-12-01 00:00:00 to:2022-12-14 16:13:00'}, index=[0]): 创建包含字符串数据的新DataFrame。index=[0]确保新行的索引为0。
pd.concat([new_row, df.loc[:]]): 将新行DataFrame与原始DataFrame合并。df.loc[:]确保选取所有行，避免潜在的SettingWithCopyWarning。
.reset_index(drop=True): 重置索引，丢弃旧索引，生成新的连续索引。
df.to_excel(writer, sheet_name="foglio1", startrow=1, header=False, index=False): 将DataFrame写入Excel文件。startrow=1表示从第二行开始写入数据，留出第一行给表头。header=False和index=False表示不写入列名和索引。
worksheet.add_table(0, 0, max_row, max_col - 1, {"columns": column_settings}): 在Excel中添加表格，设置列标题。
worksheet.set_column(0, max_col - 1, 70): 设置列宽。