前言
前几天在学习Python自动化书本中案例的时候,偶然想对数据分列多一些操作,但是遇到了问题,如下图所示。
上图这个是原始数据,但是现在想要下图这样的效果,怎么破呢?
这个问题竟然在网上找了很久,没有找到合适的,也许是我问问题的没有问到点子上,不过还好比较幸运,在群里大佬给出了思路和答案。
一、思路
一开始群友想到的是使用Excel进行分列,这个操作我自然熟悉了,只不过列是分割了,但是其他的行数据没有一起跟过来,如果你说大不了复制粘贴呗,也花不了几秒钟,我觉得也是没毛病的,这也确实是一直思路,不死磕也行。
不过不要慌,问题不大,这里给出【1px】大佬给出的解决方法,拍案惊奇!
二、解决方案
针对该问题,其实有两个方法,第一个是书中给出的openpyxl库进行拆解,如下图所示:
第二个是使用pandas中的explode()函数,这里直接给出【1px】大佬答案,如下图所示:
其实关键点就是pandas中的爆炸函数explode(),早在之前我看到过有人用这个,只是一直不知道怎么用,今天在这里算是涨知识了。
import pandas as pd
df = pd.read_excel('keywords.xlsx')
# ['序号', '年份', '来源出版物名称', '索引关键字'
df.columns
df.loc[:, ['索引关键字']] = df['索引关键字'].str.split(";") # expand=True 可以把用分割的内容直接分列
df.head()
df.explode(column='索引关键字')
最后得到预取的效果如下图所示:
上面我只是拿一行数据进行测试,接下来使用多行进行测试看看:
代码运行之后,发现都可以满足要求:
三、总结
本文基于实际过程中遇到的Excel数据拓展分列的问题,使用pandas中的explode()函数顺利完成解答,一个小题目,帮助自己和大家加深对该函数的认识。