0

0

使用 Pandas 处理包含列表的 DataFrame 列合并问题

花韻仙語

花韻仙語

发布时间:2025-07-02 16:22:20

|

1029人浏览过

|

来源于php中文网

原创


本文将围绕以下步骤展开,解决如何合并 DataFrame 中具有成对属性名称和属性值的列,特别是当属性值列可能包含列表的情况:

  1. 提取列名信息:利用 Pandas 的 extract 函数从列名中提取关键信息,构建 MultiIndex。
  2. 重塑 DataFrame:使用 stack 函数将 DataFrame 重塑为更适合处理的格式。
  3. 重命名列并添加前缀:为了清晰起见,重命名列并为列名添加 "Attribute " 前缀。
  4. 分割属性值:使用 split 函数将包含逗号分隔值的属性值列分割成列表。
  5. 展开列表:使用 explode 函数将列表展开为单独的行,从而得到最终的 DataFrame。

详细步骤及代码示例

假设我们有以下 DataFrame,需要将其转换为目标格式:

import pandas as pd

data = {'Title': ['Title 1', 'Title 2'],
        'Description': ['Desc 1', 'Desc 2'],
        'Attribute 1 name': ['Sport', 'Size'],
        'Attribute 1 value': ['NFL', 'Large, Medium'],
        'Attribute 2 name': ['Sport', 'Sleeve Type'],
        'Attribute 2 value': ['NBA', 'Long Sleeve, Short Sleeve']}
df = pd.DataFrame(data)

print(df)

输出:

    Title Description Attribute 1 name Attribute 1 value Attribute 2 name  \
0  Title 1     Desc 1             Sport               NFL             Sport   
1  Title 2     Desc 2              Size       Large, Medium       Sleeve Type   

  Attribute 2 value  
0               NBA  
1  Long Sleeve, Short Sleeve  

1. 提取列名信息

首先,使用 str.extract 函数从列名中提取数字和属性类型(name 或 value),并创建一个 MultiIndex。正则表达式 r'\S+ (\d+) (\S+)' 用于匹配列名中的模式,其中 \S+ 匹配一个或多个非空白字符,(\d+) 匹配一个或多个数字,(\S+) 匹配一个或多个非空白字符。

idx = pd.MultiIndex.from_frame(
    df.columns
      .str.extract(r'\S+ (\d+) (\S+)')                                
)
print(idx)

输出:

MultiIndex([(   '1',  'name'),
            (   '1', 'value'),
            (   '2',  'name'),
            (   '2', 'value')],
           )

2. 重塑 DataFrame

使用 set_axis 函数将创建的 MultiIndex 应用于 DataFrame 的列,然后使用 stack 函数将 DataFrame 从宽格式转换为长格式。stack(0) 将 MultiIndex 的第一层(即数字)堆叠到行索引中。

微信 WeLM
微信 WeLM

WeLM不是一个直接的对话机器人,而是一个补全用户输入信息的生成模型。

下载
out = (df.set_axis(idx, axis=1).stack(0)
         .rename_axis(columns=None)
         .add_prefix('Attribute ')
         .sort_index(level=-1)
      )

print(out)

输出:

  Attribute name Attribute value
0          Sport             NFL
1           Size   Large, Medium
0          Sport             NBA
1    Sleeve Type  Long Sleeve, Short Sleeve

3. 分割属性值并展开列表

使用 str.split(', *') 将 "Attribute value" 列中的逗号分隔值分割成列表。然后,使用 explode 函数将列表展开为单独的行。explode('Attribute value', ignore_index=True) 将 "Attribute value" 列中的每个列表展开为多行,并重置索引。

out = (df.set_axis(idx, axis=1).stack(0)
         .rename_axis(columns=None)
         .add_prefix('Attribute ')
         .sort_index(level=-1)
         .assign(**{'Attribute value': lambda d: d['Attribute value'].str.split(', *')})
         .explode('Attribute value', ignore_index=True)
      )

print(out)

输出:

  Attribute name Attribute value
0          Sport             NFL
1           Size           Large
2           Size          Medium
3          Sport             NBA
4    Sleeve Type     Long Sleeve
5    Sleeve Type    Short Sleeve

总结

通过结合使用 extractstacksplitexplode 等 Pandas 函数,我们可以有效地处理包含列表的 DataFrame 列合并问题。这种方法不仅简洁,而且易于理解和维护。

注意事项:

  • 在实际应用中,请根据实际数据调整正则表达式 r'\S+ (\d+) (\S+)',以确保正确提取列名信息。
  • 如果 DataFrame 包含大量列,stack 操作可能会消耗大量内存。在这种情况下,可以考虑使用分块处理或其他优化技术。
  • 在调用 explode 之前,请确保要展开的列确实包含列表。否则,explode 操作可能会导致意外的结果。

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
js正则表达式
js正则表达式

php中文网为大家提供各种js正则表达式语法大全以及各种js正则表达式使用的方法,还有更多js正则表达式的相关文章、相关下载、相关课程,供大家免费下载体验。

506

2023.06.20

正则表达式不包含
正则表达式不包含

正则表达式,又称规则表达式,,是一种文本模式,包括普通字符和特殊字符,是计算机科学的一个概念。正则表达式使用单个字符串来描述、匹配一系列匹配某个句法规则的字符串,通常被用来检索、替换那些符合某个模式的文本。php中文网给大家带来了有关正则表达式的相关教程以及文章,希望对大家能有所帮助。

246

2023.07.05

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

722

2023.07.05

java正则表达式匹配字符串
java正则表达式匹配字符串

在Java中,我们可以使用正则表达式来匹配字符串。本专题为大家带来java正则表达式匹配字符串的相关内容,帮助大家解决问题。

209

2023.08.11

正则表达式空格
正则表达式空格

正则表达式空格可以用“s”来表示,它是一个特殊的元字符,用于匹配任意空白字符,包括空格、制表符、换行符等。本专题为大家提供正则表达式相关的文章、下载、课程内容,供大家免费下载体验。

343

2023.08.31

Python爬虫获取数据的方法
Python爬虫获取数据的方法

Python爬虫可以通过请求库发送HTTP请求、解析库解析HTML、正则表达式提取数据,或使用数据抓取框架来获取数据。更多关于Python爬虫相关知识。详情阅读本专题下面的文章。php中文网欢迎大家前来学习。

293

2023.11.13

正则表达式空格如何表示
正则表达式空格如何表示

正则表达式空格可以用“s”来表示,它是一个特殊的元字符,用于匹配任意空白字符,包括空格、制表符、换行符等。想了解更多正则表达式空格怎么表示的内容,可以访问下面的文章。

229

2023.11.17

正则表达式中如何匹配数字
正则表达式中如何匹配数字

正则表达式中可以通过匹配单个数字、匹配多个数字、匹配固定长度的数字、匹配整数和小数、匹配负数和匹配科学计数法表示的数字的方法匹配数字。更多关于正则表达式的相关知识详情请看本专题下面的文章。php中文网欢迎大家前来学习。

526

2023.12.06

excel制作动态图表教程
excel制作动态图表教程

本专题整合了excel制作动态图表相关教程,阅读专题下面的文章了解更多详细教程。

30

2025.12.29

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号