如何用Python将DataFrame导出至Excel并添加1-9范围的数据验证列

首先确保安装所需依赖：

pip install pandas openpyxl

以下是完整实现代码，包含示例DataFrame、导出、添加数据验证列的全过程：

import pandas as pd
from openpyxl import load_workbook
from openpyxl.worksheet.datavalidation import DataValidation

# 构建示例DataFrame（替换为你的实际数据）
df = pd.DataFrame({
    'name': ['Alice', 'Bob', 'Charlie'],
    'year': [2021, 2022, 2023]
})

# 导出DataFrame到Excel，指定openpyxl引擎以便后续编辑
output_path = 'data_with_code.xlsx'
df.to_excel(output_path, index=False, engine='openpyxl')

# 加载已导出的Excel文件
wb = load_workbook(output_path)
ws = wb.active

# 添加code列的表头
ws['C1'] = 'code'

# 创建数据验证规则：仅允许输入1-9的整数
code_validation = DataValidation(
    type="whole",
    operator="between",
    formula1=1,
    formula2=9,
    allow_blank=False,
    # 配置错误提示
    showErrorMessage=True,
    errorTitle="输入无效",
    error="请输入1-9之间的整数",
    # 配置输入提示
    showInputMessage=True,
    promptTitle="输入提示",
    prompt="请填写1-9的数字"
)

# 将验证规则应用到C列的有效数据区域（从第2行开始，这里设置到C1000适配后续新增行）
code_validation.add('C2:C1000')
ws.add_data_validation(code_validation)

# 保存修改后的文件
wb.save(output_path)

关键说明： #

• 选择openpyxl作为Excel引擎，因为它支持对已生成的Excel文件进行格式和数据验证的编辑操作
• 数据验证规则限制输入为1-9的整数，同时配置了输入提示和错误提示，优化用户填写体验
• 验证范围设置为C2:C1000，既覆盖现有DataFrame的行，也支持后续手动新增行时自动应用规则，可根据实际需求调整范围

内容的提问来源于stack exchange，提问作者codingrainha