Python学习笔记: Pandas pipe 及 get_dummies 範例

本文同步发表于小弟自架网站：微确幸资讯站

原始资料：

武功简历一灯大师一阳指，先天功，九阴真经杨过全真剑法，弹指神通，九阴真经，黯然销魂掌王重阳全真剑法，先天功，一阳指黄药师落英神剑掌，弹指神通

将武功简历中有学过的标示为「1」，没学过的标示为「0」。

预期结果：

武功简历一阳指九阴真经先天功全真剑法弹指神通落英神剑掌黯然销魂掌一灯大师一阳指，先天功，九阴真经1110000杨过全真剑法，弹指神通，九阴真经，黯然销魂掌0101101王重阳全真剑法，先天功，一阳指1011000黄药师落英神剑掌，弹指神通0000110

import pandas as pddf = pd.DataFrame({'武功简历': {'杨过': '全真剑法，弹指神通，九阴真经，黯然销魂掌',  '黄药师': '落英神剑掌，弹指神通',  '王重阳': '全真剑法，先天功，一阳指',  '一灯大师': '一阳指，先天功，九阴真经'}}  )df

# pandas explode 配合 pipe 及 get_dummies后的状况df['武功简历'].str.split('，').explode().pipe(pd.get_dummies)

# 最终程式以人物 groupby 后，对所有的武功做加总得到结果df.join(df['武功简历'].str.split('，').explode().pipe(pd.get_dummies).groupby(level=0).sum())