如何使用if else语句将数组中元素的变化值写入excel？

import openpyxl

# 打开Excel文件
workbook = openpyxl.load_workbook('example.xlsx')

# 获取要写入的工作表对象
worksheet = workbook['Sheet1']

# 数组中的元素
array = [1, 2, 3, 4, 5]

# 遍历数组中的元素
for i, value in enumerate(array):
    # 使用if else语句判断元素的值
    if value < 3:
        # 将变化值写入Excel中的相应单元格
        worksheet.cell(row=i+1, column=2).value = value + 1
    else:
        worksheet.cell(row=i+1, column=2).value = value - 1

# 保存并关闭Excel文件
workbook.save('example.xlsx')
workbook.close()