Python基礎教程：一個單列split轉換為多行的練習題

一、問題提出

實作如圖所示的轉換。

姓名	年齡
A	15
B、C	16

B
C

二、解決問題

'''
學習中遇到問題沒人解答？小編建立了一個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
'''
           

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
'''