如何用更Pythonic的方式拼接Pandas DataFrame指定列内容?
高效Python风格实现方案
你原来的嵌套循环逻辑在数据量较大时性能会非常差,pandas提供了原生向量化操作可以完美替代循环,以下是两种适配不同场景的实现:
方案1:适配原逻辑(遇到空值/非字符串值停止拼接)
和你现有脚本逻辑完全一致,行迭代过程中如果遇到空值或者非字符串内容就停止拼接后续列,代码更简洁易读:
import pandas as pd # 筛选需要拼接的列:对应你原逻辑的第1列到倒数第二列 # 如果只需要排除索引为0的第一列,不需要排除最后一列,改成cols_to_concat = df_dt_proc.columns[1:]即可 cols_to_concat = df_dt_proc.columns[1:-1] def concat_row(row): res = [] for val in row: if pd.notna(val): res.append(str(val)) else: break return ', '.join(res) df_dt_proc['procedures'] = df_dt_proc[cols_to_concat].agg(concat_row, axis=1)
方案2:全量拼接(性能最优)
如果你不需要遇到空值就停止,仅需要把所有非空的列值用, 拼接,这个方法底层用C实现,完全避免纯Python层的行迭代,性能比嵌套循环高10~100倍,数据量越大优势越明显:
cols_to_concat = df_dt_proc.columns[1:-1] # 所有列转字符串,空值替换为空字符串 df_process_cols = df_dt_proc[cols_to_concat].astype(str).replace('nan', '') # 按行拼接 df_dt_proc['procedures'] = df_process_cols.agg(', '.join, axis=1)
内容的提问来源于stack exchange,提问作者gm-123
相关产品推荐
相关产品推荐

