Écrire des données multi-indexées dans un fichier Excel avec Python/Pandas

Question

Je souhaite créer une feuille de calcul Excel et insérer un nombre égal de lignes pour chaque variable. Le résultat idéal doit ressembler à Colonnes A & B dans l'image. Ce que je peux faire jusqu'ici est seulement d'insérer pour 1 nom (Colonnes D & E), et n'ai aucune idée faire l'énumération appropriée pour le reste.

C'est ce que j'ai:

import xlwt, xlrd 
import os 

current_file = xlwt.Workbook() 
write_table = current_file.add_sheet('Sheet1') 

name_list = ["Jack", "David", "Andy"] 
food_list = ["Ice-cream", "Mango", "Apple", "Cake"] 

total_rows = len(name_list) * len(food_list) # how to use it? 

write_table.write(0, 0, "Jack") 

for row, food in enumerate(food_list): 
    write_table.write(row, 1, food) 

current_file.save("c:\\name_food.xls") 

Comment puis-je faire pour tous? Je vous remercie.

Answer 1

Quelque chose comme cela devrait fonctionner:

import xlwt, xlrd 
import os 

current_file = xlwt.Workbook() 
write_table = current_file.add_sheet('Sheet1') 

name_list = ["Jack", "David", "Andy"] 
food_list = ["Ice-cream", "Mango", "Apple", "Cake"] 

for i, name in enumerate(name_list): 
    write_table.write(i * len(food_list), 0, name_list[i]) 

    for row, food in enumerate(food_list): 
     write_table.write(i * len(food_list) + row, 1, food) 

current_file.save("c:\\name_food.xls") 

L'important est

write_table.write(i * len(food_list), 0, name_list[i]) 

où vous dites que le nom doit être écrit en lignes 0, 4, 8, 12 ...

En outre, une partie

write_table.write(i * len(food_list) + row, 1, food) 

écrit la nourriture à la section correspondante augmentée par le numéro de rangée.

Answer 2

Vous pouvez créer DataFrame par numpy.tile et numpy.repeat puis retirez dupliqués dans a colonne:

df = pd.DataFrame({'a': np.repeat(name_list, len(food_list)), 
        'b': np.tile(food_list, len(name_list))}) 

df['a'] = np.where(df['a'].duplicated(), '', df['a']) 
print (df) 
     a   b 
0 Jack Ice-cream 
1    Mango 
2    Apple 
3    Cake 
4 David Ice-cream 
5    Mango 
6    Apple 
7    Cake 
8 Andy Ice-cream 
9    Mango 
10    Apple 
11    Cake 

Une autre solution avec la compréhension de la liste:

df = pd.DataFrame({'a': [y for x in name_list for y in [x] + [''] * (len(food_list)-1)], 
        'b': food_list * len(name_list)}) 
print (df) 
     a   b 
0 Jack Ice-cream 
1    Mango 
2    Apple 
3    Cake 
4 David Ice-cream 
5    Mango 
6    Apple 
7    Cake 
8 Andy Ice-cream 
9    Mango 
10    Apple 
11    Cake 

et dernière écriture to_excel:

df.to_excel('c:\\name_food.xls', index=False, header=False) 

Answer 3

En utilisant pandas 'pd.MultiIndex.from_product:

df = pd.DataFrame(index=pd.MultiIndex.from_product([name_list, food_list])) 
df.to_excel("c:\\name_food.xls", sheet_name="Sheet1") 

Le fichier Excel contient maintenant:

Jack Ice-cream 
    Jack  Mango 
    Jack  Apple 
    Jack  Cake 
David Ice-cream 
David  Mango 
David  Apple 
David  Cake 
    Andy Ice-cream 
    Andy  Mango 
    Andy  Apple 
    Andy  Cake 

---

Si vous voulez que le format exact comme indiqué dans votre image, vous pouvez réinitialiser l'index, puis convertir lignes dupliquées pour vider des chaînes.

df = pd.DataFrame(index=pd.MultiIndex.from_product(\ 
           [name_list, food_list])).reset_index() 
df['level_0'].loc[df['level_0'].duplicated()] = '' 
df.to_excel("c:\\name_food.xls", sheet_name="Sheet1", index=False, columns=False) 

Le fichier Excel ressemblera à ceci:

 Jack Ice-cream 
       Mango 
       Apple 
       Cake 
    David Ice-cream 
       Mango 
       Apple 
       Cake 
    Andy Ice-cream 
       Mango 
       Apple 
       Cake