Python实现根据Excel表格某一列内容与数据库进行匹配

pip install pandas brightway2

import pandas as pd
import bw2data as bd

def match_and_export(excel_file_path, db_name, match_column_name, match_condition):
    # 读取Excel文件
    df = pd.read_excel(excel_file_path)
    # 获取匹配列的数据
    match_values = df[match_column_name].tolist()

    # 加载Brightway2数据库
    bd.projects.set_current('your_project_name')  # 设置当前项目名称，需要替换为实际的项目名称
    db = bd.Database(db_name)

    # 用于存储匹配成功的数据
    matched_data = []

    # 遍历Excel中的匹配值
    for value in match_values:
        # 根据匹配条件进行匹配
        for activity in db:
            if match_condition(activity, value):
                # 这里可以根据需要抓取更多的信息
                data = {
                    'excel_value': value,
                    'activity_name': activity['name'],
                    'activity_code': activity['code']
                }
                matched_data.append(data)

    # 将匹配成功的数据转换为DataFrame
    result_df = pd.DataFrame(matched_data)

    # 导出为新的Excel文件
    result_df.to_excel('matched_data.xlsx', index=False)
    print("匹配数据已导出到 matched_data.xlsx")

# 示例匹配条件函数
def example_match_condition(activity, value):
    # 这里假设根据活动名称进行匹配，你可以根据实际需求修改
    return activity['name'] == value

# 示例调用
if __name__ == "__main__":
    excel_file_path = 'your_excel_file.xlsx'  # 替换为实际的Excel文件路径
    db_name = 'your_database_name'  # 替换为实际的数据库名称
    match_column_name = 'your_column_name'  # 替换为实际的匹配列名称

    match_and_export(excel_file_path, db_name, match_column_name, example_match_condition)