pandas 把一个含有数组或列表的列扩展成多行的函数 explode

最新推荐文章于 2024-09-22 17:12:26 发布

三梦行者

最新推荐文章于 2024-09-22 17:12:26 发布

阅读量1.1k

点赞数

文章标签： python

本文链接：https://blog.csdn.net/leishensuzi/article/details/111289578

版权

本文介绍了如何使用pandas DataFrame对包含嵌套列表的df_dict进行explode操作，随后展示了如何利用pivot_table进行数据转换，以创建一个按'a'列和'c'列分组的值矩阵。最终输出了带填充值的pivot表并展示了结果。

摘要由CSDN通过智能技术生成

import pandas as pd

df_dict = {
    'a': ['a1','a2','a3'],
    'c': [['c1','c2'], ['c2','c3','c4'],['c2','c3']]
}

df = pd.DataFrame(df_dict)
new_df = df.explode('c')

new_df['val']=1
pivot_df = new_df.pivot_table(columns='a', index='c',values='val', fill_value=0)
print(pivot_df.reset_index())

效果：