Блог

У меня есть две таблицы A / B. Я хочу объединить таблицу слияния A в B при создании цикла.

Ввод

 Table A:            Table B:

name      year      name     subject group
James     one       Jenny    Physics
Jackson   one       Jackson  Maths
Jenny     two       Jenny    PE
Himeth    three     Himeth   Chemistry
Carlos    three     Carlos   Physics
Mendy     one       Mendy    German
James     two       James    Physics
Himeth    one       Himeth   Chemistry
 

Вывод

 Name    subject    group   year_one year_two year_three
Jenny   PE         Yellow           two     
Jackson Maths      Green   one              
James   Physics    Yellow  one      two     
Himeth  Chemistry  Yellow  one                     three
Carlos  Physics    Green                           three
Mendy   German     Yellow  one      
 

Я хочу добавить столбец year из таблицы A, но создать столбцы для year (один, два, три), поскольку есть дубликаты.

years = [«один», «два», «три»]

 for i in years: 
    df.to_csv("student.csv")
    dfs = pd.merge(df,students,
                on = "Name",
                how = "left")
    dfs[i] = i
    dfs.to_csv("class_setup.csv")
 

Я не уверен, как выполнить цикл для слияния и создать новый столбец.

Используйте DataFrame.pivot с DataFrame.join :

 df = df2.join(df1.pivot('name','year','year'), on='name').fillna('')
print (df)
      name  one  three  two
0    Jenny              two
1  Jackson  one            
2    Jenny              two
3   Himeth  one  three     
4   Carlos       three     
5    Mendy  one            
6    James  one         two
7   Himeth  one  three     
 

Если порядок важен:

 years = ["one", "two", "three"]
df = df2.join(df1.pivot('name','year','year').reindex(years, axis=1), on='name').fillna('')
print (df)
      name  one  two  three
0    Jenny       two       
1  Jackson  one            
2    Jenny       two       
3   Himeth  one       three
4   Carlos            three
5    Mendy  one            
6    James  one  two       
7   Himeth  one       three