Python使用openpyxl设置Excel单元格公式和工作簿合并

pip install openpyxl

uv add openpyxl

from openpyxl import Workbook
# 创建一个新的工作簿和活动工作表
wb = Workbook()
ws = wb.active

# 为 A1 和 B1 单元格设置数值
ws['A1'] = 10
ws['B1'] = 5

# 为 C1 单元格设置公式，计算 A1 和 B1 的和
ws['C1'] = '=SUM(A1:B1)'

# 为 D1 单元格设置另一个公式，计算 A1 除以 B1 的结果并保留两位小数
ws['D1'] = '=ROUND(A1/B1, 2)'

# 保存工作簿
wb.save('formula_example.xlsx')

print("已成功为单元格设置公式并保存到 formula_example.xlsx")

='[workbook_name.xlsx]sheet_name'!cell_reference

from openpyxl import Workbook
# 创建一个新的工作簿和活动工作表
wb = Workbook()
ws = wb.active
# 为 A1 和 B1 单元格设置数值
ws['A1'] = 10
ws['B1'] = 5
# 为 C1 单元格设置公式，引用另一个工作簿中的单元格
ws['C1'] = "='[other_workbook.xlsx]Sheet'!A1 + '[other_workbook.xlsx]Sheet'!B1"
# 保存工作簿
wb.save('formula_with_external_reference.xlsx')
print("已成功为单元格设置引用外部工作簿的公式并保存到 formula_with_external_reference.xlsx")

from openpyxl import load_workbook, Workbook
import os

def merge_sheets(source_files, target_file, sheet_index=0):
    '''
    合并多个Excel文件中的指定工作表到一个新的Excel文件中。
    参数:
        source_files: 包含源Excel文件路径的列表。
        target_file: 目标Excel文件的路径。
        sheet_index: 要合并的工作表的索引，默认为0（第一个工作表）。
    '''
    # 创建一个新的工作簿作为目标文件
    target_wb = Workbook()
    # 删除默认创建的第一个工作表，因为我们稍后会添加自己的
    if 'Sheet' in target_wb.sheetnames:
        del target_wb['Sheet']

    for file in source_files:
        if not os.path.exists(file):
            print(f"文件 {file} 不存在，跳过。")
            continue
        # 目标工作簿中创建对应的表
        target_ws = target_wb.create_sheet(title=os.path.splitext(os.path.basename(file))[0])
        # 加载源工作簿
        source_wb = load_workbook(filename=file)
        source_ws = source_wb.worksheets[sheet_index]

        # 复制源工作表的数据到目标工作表
        current_row = 1
        for row in source_ws.iter_rows(values_only=True):
            for col_index, value in enumerate(row, start=1):
                target_ws.cell(row=current_row, column=col_index, value=value)
            current_row += 1

    # 保存合并后的工作簿
    target_wb.save(target_file)
    print(f"已成功合并工作表并保存到 {target_file}")

# 示例用法
if __name__ == "__main__":
    source_files = []
    for i in range(1, 4):
        source_files.append(f'./data/快递价格{i}.xlsx')
    target_file = 'merged_sheets.xlsx'
    merge_sheets(source_files, target_file)