python - 自动拆分并生成列

Question

df

fruit                want_split
apple    colour:red, like:yes, favourite:no
banana   colour:yellow, like:yes, favourite:no
pear     colour:green, like:yes, favourite:yes

有没有办法用逗号分割，然后自动生成列名，而不是手动？预期输出:

fruit                want_split                  colour      like    favourite
apple    colour:red, like:yes, favourite:no     red        yes       no
banana   colour:yellow, like:yes, favourite:no  yellow     yes       no
pear     colour:green, like:yes, favourite:yes  green      yes       yes

Answer 1

您可以在str.split和explode之后使用pivot/unstack和join:

u = df['want_split'].str.split(",").explode().str.split(":",expand=True)
out = df.join(u.set_index(0,append=True)[1].unstack())

---

print(out)

     type                             want_split  favourite  like  colour
0   apple     colour:red, like:yes, favourite:no         no   yes     red
1  banana  colour:yellow, like:yes, favourite:no         no   yes  yellow
2    pear  colour:green, like:yes, favourite:yes        yes   yes   green