如何在Python中按条件为培训计划数据库添加行

前置准备 #

先安装依赖的pandas库，用于高效处理表格数据：

pip install pandas

完整脚本 #

import pandas as pd

# 1. 读取原始培训数据（假设是CSV格式，若为Excel可改用pd.read_excel）
df = pd.read_csv("original_training_plan.csv")

# 2. 定义需要生成的月份列表（可根据需求修改格式，比如['2024-01', '2024-02', ...]）
target_months = ['1月', '2月', '3月', '4月', '5月', '6月', '7月', '8月', '9月', '10月', '11月', '12月']

# 3. 为每条培训条目添加月份列并展开成多行
# 先给每行添加包含所有目标月份的列表列，再通过explode拆分
df['Month'] = [target_months] * len(df)
df_expanded = df.explode('Month', ignore_index=True)

# 4. 保存处理后的数据（保存为CSV，也可改用to_excel输出Excel文件）
df_expanded.to_csv("expanded_training_plan.csv", index=False, encoding='utf-8-sig')

print("数据处理完成，已保存为expanded_training_plan.csv")

关键说明 #

• 高效处理大数量：pandas的explode方法能快速处理2000+条数据，无需循环遍历，性能远超手动操作
• 自定义月份：修改target_months列表即可调整需要生成的月份，支持任意格式（如数字、带年份的字符串）
• 格式兼容：若原始数据是Excel文件，只需把pd.read_csv改成pd.read_excel("原始文件.xlsx")，保存时用to_excel即可

使用步骤 #

1. 将原始培训数据导出为CSV（或Excel）文件，命名为original_training_plan.csv放在脚本同目录
2. 调整target_months为你需要的月份集合
3. 运行脚本，生成的带月份列的扩展数据会保存在expanded_training_plan.csv中

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