Python学习笔记：一个单列split转换为多行的例子

一、问题提出

实现如图所示的转换。

二、解决问题

1.方法一：堆叠

# 构建测试数据
import pandas as pd
df = pd.DataFrame({'name':['a', 'b,c'],
                   'age':[15, 16]})
print(df)

df.set_index('age')['name'].str.split(',', expand=True).stack().reset_index(level=1, drop=True).reset_index(name='name')
'''
   age name
0   15    a
1   16    b
2   16    c
'''

2.方法二：循环

def shift(arr1, arr2):
    arr = []
    for k,v in dict(zip(arr1, arr2)).items():
        # print(k)
        # print(v)
        if isinstance(v, str):
            for s in v.split(','):
                arr.append({'name':k, 'age':s})
    return pd.DataFrame(arr)

shift(df.age, df.name)
'''
  age  name
0   a    15
1   b    16
2   c    16
'''