将“for”循环的输出写入PYTHON中的excel

时间:2018-03-14 13:33:37

标签: python pandas xlsxwriter

我有以下代码:

my_list = ["US", "IT", "ES", "NL"]
for i in my_list:
    A = sum_products_by_country(world_level,i)
    df = pd.DataFrame({'value':A})
    Descending = df.sort_values( by='value', ascending = 0 )
    Top_5 = Descending[0:5]
    print(Top_5)

“sum_products_by_country”是一个创建的函数,它将数据框(在我的例子中命名为“world_level”)和国家/地区名称作为参数,并返回此国家/地区的产品总和。使用这个循环我找到top5产品和my_list的每个国家的总和。以下是此循环的输出:

US          value
Product  


B          1492

H          455

BB         351

C          119

F          117

IT          value
Product


P           346
U           331

A           379

Q           190

D          1389

ES         value
Product 


P          3046

U3         331

A          379

Q          1390

DD         10389

NL         value
Product 


P          3465

U          3313

AA         379

2Q         190

D          189

我想使用以下方法在excel表中编写此输出:

writer = pd.ExcelWriter('top products.xlsx', engine='xlsxwriter')
Top_5.to_excel(writer, sheet_name='Sheet1')
writer.save()

你能告诉我在哪里可以输入上面的代码来获取所需的excel文件吗? 是否还有一种方法可以将列名(国家/地区,产品,价值)仅在我的Excel文档的顶部获取一次而不是分别用于每个国家/地区?所以我想要这样的东西:

 Country   Product   value

  US        
           B          1492
           H          455
           BB         351
           C          119
           F          117

  IT          

           P           346
           U           331
           A           379
           Q           190
           D          1389

  ES         

          P          3046
          U3         331
          A          379
          Q          1390
          DD         10389

  NL         

          P          3465
          U          3313
          AA         379
          2Q         190
          D          189

谢谢

1 个答案:

答案 0 :(得分:0)

此脚本可以帮助您:

#Create workbook object
wb = openpyxl.Workbook()
sheet = wb.get_active_sheet()
sheet.title='Products by country'

#Generate data

#Add titles in the first row of each column
sheet.cell(row=1, column=1).value='country'
sheet.cell(row=1, column=2).value='product'
sheet.cell(row=1, column=3).value='value'


#Loop to set the value of each cell
for i in range(0, len(Country)):
sheet.cell(row=i+2, column=1).value=Country[i]#with country being your array full of country names. If you have 5 values for one country I would advise just having the country name in there five times.
sheet.cell(row=i+2, column=2).value=Product[i]#array with products
sheet.cell(row=i+2, column=3).value=Values[i]#array with values

#Finally, save the file and give it a name
wb.save('NameFile.xlsx')