如何在Python中实现Excel相对引用公式

from openpyxl import Workbook
from openpyxl.utils import get_column_letter
from openpyxl.formula import Tokenizer, Token

def get_relative_formula(formula, row_offset, col_offset):
    tokenizer = Tokenizer(formula)
    tokens = tokenizer.items
    new_tokens = []
    
    for token_type, token_value in tokens:
        if token_type == Token.OP:
            new_tokens.append((token_type, token_value))
        elif token_type == Token.CELL:
            row, col = tokenizer.parse_cell(token_value)
            new_row = row + row_offset
            new_col = col + col_offset
            new_cell = get_column_letter(new_col) + str(new_row)
            new_tokens.append((Token.CELL, new_cell))
    
    new_formula = ''.join(token[1] for token in new_tokens)
    return new_formula

# 创建一个新的Excel文件
wb = Workbook()
ws = wb.active

# 设置一些示例数据
ws['A1'] = 1
ws['B1'] = 2
ws['C1'] = 3

# 设置公式
ws['A2'] = '=A1 + B1'
ws['B2'] = '=B1 + C1'
ws['C2'] = '=C1 + A1'

# 获取相对引用公式
new_formula = get_relative_formula(ws['A2'].value, 1, 1)
ws['A3'] = new_formula

new_formula = get_relative_formula(ws['B2'].value, 1, 1)
ws['B3'] = new_formula

new_formula = get_relative_formula(ws['C2'].value, 1, 1)
ws['C3'] = new_formula

# 保存Excel文件
wb.save('example.xlsx')