表格中提取个别“行名称”所对应的数据并输出的python代码

第一部分：需求

我想要提取1号和3号这两行并输出xlsx文件

第二部分：代码实现

import pandas as pd

# 定义文件路径
input_file_path = 'before.xlsx'  # 输入文件路径
output_file_path = 'result.xlsx'  # 输出文件路径
sheet_name = 'Sheet1'  # 数据所在的工作表名称

# 读取指定工作表中的数据
data = pd.read_excel(input_file_path, sheet_name=sheet_name)

# 提取“序号”为“1号”和“3号”的行数据
row_data = data[data['序号'].isin(['1号', '3号'])]

# 将提取的数据保存为新的Excel文件
row_data.to_excel(output_file_path, index=False)

print(f"提取的数据已保存至 {output_file_path}")

第三部分：效果

注意点：

如果提取失败，代表第一列有空格，在row_data前面加上：

# 去除 "序号" 列中的前后空格，以确保匹配准确
data['序号'] = data['序号'].str.strip()