如何用Python按客户提取Excel数据？求推荐适用库及方法

推荐工具库 #

处理这类Excel数据提取需求，pandas是最便捷的选择——它专门用于表格数据的读取、处理和导出，配合openpyxl（支持.xlsx格式）或xlrd（支持旧版.xls格式）作为读取引擎即可完成任务。

步骤1：安装依赖库 #

在终端执行以下命令安装所需包：

pip install pandas openpyxl

如果你的Excel文件是旧版.xls格式，需额外安装兼容版xlrd：

pip install xlrd==1.2.0

步骤2：核心代码实现 #

下面是可直接复用的代码，包含输入客户名称、校验工作表存在性、提取指定数据的逻辑：

import pandas as pd

def get_customer_data(excel_file_path, customer_name, target_columns):
    # 加载Excel文件并获取所有工作表名称
    excel_handler = pd.ExcelFile(excel_file_path, engine='openpyxl')
    available_sheets = excel_handler.sheet_names
    
    # 检查目标客户工作表是否存在
    if customer_name not in available_sheets:
        print(f"未找到客户「{customer_name}」对应的工作表")
        return None
    
    # 读取指定工作表的指定列数据
    customer_data = excel_handler.parse(customer_name, usecols=target_columns)
    return customer_data

# ------------------- 使用示例 -------------------
if __name__ == "__main__":
    # 替换为你的Excel文件路径
    file_path = "客户总报表.xlsx"
    # 接收用户输入的客户名称
    input_customer = input("请输入要查询的客户名称：")
    # 定义需要提取的列（根据你的实际需求修改）
    columns_need = ["订单编号", "交易金额", "交易日期"]
    
    extracted_data = get_customer_data(file_path, input_customer, columns_need)
    if extracted_data is not None:
        print("\n提取到的数据：")
        print(extracted_data)
        # 可选：将提取结果保存为新的Excel文件
        extracted_data.to_excel(f"{input_customer}_数据提取结果.xlsx", index=False, engine='openpyxl')

额外说明 #

• 如果需要提取行级筛选数据（比如只提取某日期后的订单），可以在读取数据后添加过滤逻辑：

  # 示例：筛选交易金额大于1000的记录
  filtered_data = extracted_data[extracted_data["交易金额"] > 1000]
  

• 若需更精细的单元格操作（比如读取特定单元格值），可以单独使用openpyxl库，但对于批量数据提取，pandas的效率更高。

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