Python基础教程：一个单列split转换为多行的练习题

一、问题提出

实现如图所示的转换。

姓名	年龄
A	15
B、C	16

用Python转换为：

姓名	年龄
A	15
B	16
C	16

二、解决问题

1.方法一：堆叠

'''
学习中遇到问题没人解答？小编创建了一个Python学习交流QQ群：531509025
寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和PDF电子书！
'''
# 构建测试数据
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
'''