使用Pandas为DataFrame指定外列增量式命名为col1、col2、col3格式

• 第一步：筛选需要重命名的列
  先定义要保留原名的列列表，再从DataFrame的列名里排除这些列，得到待重命名的列集合：

  # 指定保留原名的列
  keep_cols = ['category', 'week', 'created_date', 'year']
  # 筛选出需要重命名的列
  rename_cols = [col for col in df.columns if col not in keep_cols]
  

• 第二步：构建列名映射字典
  给待重命名的列按顺序生成col1、col2这类序列名称，创建原列名到新列名的映射关系：

  # 生成新列名并构建映射字典
  name_mapping = {old_col: f'col{i+1}' for i, old_col in enumerate(rename_cols)}
  

• 第三步：执行重命名操作
  调用Pandas的rename()方法完成列名替换：

  # 完成列重命名
  df = df.rename(columns=name_mapping)
  

• 示例效果
  假设初始DataFrame列名为['category', 'week', 'created_date', 'year', 'value', '2023', '2024']，执行代码后列名将变为['category', 'week', 'created_date', 'year', 'col1', 'col2', 'col3']。

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