pandas如何循环拼接1开头列与2开头列生成列表型输出列

实现代码 #

import pandas as pd
import re

# 筛选目标列
cols1 = [col for col in df.columns if col.startswith('1')]
cols2 = [col for col in df.columns if col.startswith('2')]

def process_row(row):
    # 清洗1开头列的有效元素
    valid_1 = []
    for col in cols1:
        val = row[col]
        if pd.isna(val):
            continue
        # 处理浮点数后缀.0，转字符串
        val_str = str(val).rstrip('.0') if isinstance(val, float) else str(val)
        # 仅保留字母数字
        val_clean = re.sub(r'\W', '', val_str)
        valid_1.append(val_clean)
    
    # 清洗2开头列的有效元素
    valid_2 = []
    for col in cols2:
        val = row[col]
        if pd.isna(val):
            continue
        val_str = str(val).rstrip('.0') if isinstance(val, float) else str(val)
        val_clean = re.sub(r'\W', '', val_str)
        valid_2.append(val_clean)
    
    # 不满足生成条件返回空
    if not valid_1 or not valid_2:
        return pd.NA
    
    # 两两拼接生成结果列表
    return [f"{v1}|{v2}" for v1 in valid_1 for v2 in valid_2]

# 应用到数据集生成Output列
df['Output'] = df.apply(process_row, axis=1)

逻辑说明 #

• 无需硬编码列名，自动识别所有1、2开头的列做处理，后续列数增减不用改核心逻辑
• 按行处理逻辑直观，完全匹配所有规则要求：自动过滤空值、清洗非字母数字字符、仅当两类列都存在有效元素时才生成拼接列表，否则返回空值
• 兼容数值型、字符串型列的处理，自动去除浮点数的.0后缀

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