pandas excel动态条件过滤并保存结果

# 查询条件，多个条件，用逗号分隔
where_dict = {
    # excel文件名
    "file_name": "456.xlsx",
    # 过滤条件
    "rules": [
        {
            "sheet_name": "Sheet1",
            "split_rule": ["性别=男", "年龄=21"]
        },
        {
            "sheet_name": "Sheet2",
            "split_rule": ["身高=170"]
        }
    ]
}

pip3 install pandas openpyxl

# ！/usr/bin/python3
# -*- coding: utf-8 -*-
import pandas as pd

# 查询条件，多个条件，用逗号分隔
where_dict = {
    # excel文件名
    "file_name": "456.xlsx",
    # 过滤条件
    "rules": [
        {
            "sheet_name": "Sheet1",
            "split_rule": ["性别=男", "年龄=21"]
        },
        {
            "sheet_name": "Sheet2",
            "split_rule": ["身高=170"]
        }
    ]
}

# 创建新的新的查询结果excel
with pd.ExcelWriter('result.xlsx') as writer:
    file_name = where_dict['file_name']
    for i in where_dict['rules']:
        # print(i, type(i))
        # 查询条件
        where = ""
        # 打开指定的sheet
        df = pd.read_excel(file_name, sheet_name=i['sheet_name'])
        for j in i['split_rule']:
            # 切割条件
            key, value = j.split('=')

            # 判断数值是否为数字
            if value.isdigit():
                # 拼接查询条件
                where += "(df.%s==%s) & " % (key, value)
            else:
                where += "(df.%s=='%s') & " % (key, value)

        # 去除末尾的&
        where = where.rstrip('& ')
        print(i['sheet_name'],'条件:',where)
        ret = df[eval(where)]
        # print(ret,type(ret))

        # 依次写入sheet
        ret.to_excel(writer, sheet_name=i['sheet_name'], index=False, header=True)

Sheet1 条件: (df.性别=='男') & (df.年龄==21)
Sheet2 条件: (df.身高==170)