pandas习题 073：展平操作列表

看过来

《pandas 教程》持续更新中，提供建议、纠错、催更等加作者微信: gr99123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。跟作者学习，请进入 Python学习课程。欢迎关注作者出版的书籍：《深入浅出Pandas》和《Python之光》。

（编码题）在 Python 中，将嵌套列表（如 a）中的内部一级数据合并为一个扁平化的列表（如 b），通常称为 “展平” 操作。想用 pandas 得到 b，如何实现？

# 有一个数据类似
a = [[['RD','red','红色'],
      ['BU','blue','蓝色']], 
     [['BK','black','黑色'],
      ['WH','white','白色']]
    ]
a
'''
[[['RD', 'red', '红色'], ['BU', 'blue', '蓝色']],
 [['BK', 'black', '黑色'], ['WH', 'white', '白色']]]
'''

期望结果为：

# 想把内部一级数据合并掉，得到
b = [['RD','red','红色'],
     ['BU','blue', '蓝色'], 
     ['BK','black','黑色'],
     ['WH','white','白色']
    ]
b
'''
[['RD', 'red', '红色'],
 ['BU', 'blue', '蓝色'],
 ['BK', 'black', '黑色'],
 ['WH', 'white', '白色']]
'''

参考答案

以下为 Python 代码：

import pandas as pd

s = pd.Series(a)
s
'''
0       [[RD, red, 红色], [BU, blue, 蓝色]]
1    [[BK, black, 黑色], [WH, white, 白色]]
dtype: object
'''

s.explode().to_list()
'''
[['RD', 'red', '红色'],
 ['BU', 'blue', '蓝色'],
 ['BK', 'black', '黑色'],
 ['WH', 'white', '白色']]
'''

解析

查看相关链接中的知识。

如果不用 Python 三方库，见 Python习题 254：展平一个嵌套列表。

（完）

pandas习题 073：展平操作列表

参考答案

解析

相关内容