如何在Pandas中通用地将一行拆分为多行?
解决DataFrame多列通用拆分爆炸的问题
你遇到的问题核心是动态列名的访问和赋值:
- 原代码里的
df.column是固定访问名为column的列,而非循环变量column对应的目标列,得改成df[column] assign的关键字参数不能直接用变量,要通过字典解包的方式动态传递列名和对应值
通用写法代码
import pandas as pd # 示例DataFrame df = pd.DataFrame({ 'name': ['a|b'], 'text': ['a test'], 'group': ['m|l|n'] }) columns = ['name', 'text', 'group'] delimiter = '|' for column in columns: if column in df.columns: # 用字典解包实现动态列赋值 df = df.assign(**{column: df[column].str.split(delimiter)}).explode(column) print(df)
运行结果
name text group 0 a a test m 0 a a test l 0 a a test n 0 b a test m 0 b a test l 0 b a test n
关键部分解释
df[column]:通过方括号+变量的方式,动态访问当前循环到的列,替代固定的df.name这类写法**{column: df[column].str.split(delimiter)}:创建以循环变量column为键的字典,用**解包后传给assign,实现动态给对应列赋值拆分后的列表explode(column):直接用循环变量指定要展开的列,不用写死列名
这样不管你要处理多少列,只要加到columns列表里就行,不用再写一堆判断列名的if语句。
内容的提问来源于stack exchange,提问作者marlon
相关产品推荐
相关产品推荐

