Python - Openpyxl -将公式添加到列，并将公式重复到数据的最后一行

from openpyxl import Workbook
from openpyxl.utils import get_column_letter
from openpyxl.styles import Font

wb = Workbook()
ws = wb.active

data = [
    ['Name', 'Age', 'Salary'],
    ['John', 25, 5000],
    ['Alice', 30, 6000],
    ['Bob', 35, 7000]
]

for row in data:
    ws.append(row)

# 假设要在第4列（D列）添加公式
column_index = 4
column_letter = get_column_letter(column_index)

# 设置公式
formula = "=B2+C2"  # 以B列和C列相加为例

# 将公式应用到指定列的每一行
for row in range(2, ws.max_row + 1):
    cell = f"{column_letter}{row}"
    ws[cell].value = formula

# 设置公式单元格的字体样式
font = Font(color="FF0000")  # 设置为红色
ws[f"{column_letter}2:{column_letter}{ws.max_row}"].font = font

wb.save("output.xlsx")