Advertisement
Guest User

Untitled

a guest
Apr 27th, 2017
81
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.66 KB | None | 0 0
  1. import argparse
  2. import pandas as pd
  3. import numpy as np
  4.  
  5. def create_pivot(infile, index_list=["Manager", "Rep", "Product"],
  6. value_list=["Price", "Quantity"]):
  7. """
  8. Read in the Excel file, create a pivot table and return it as a DataFrame
  9. """
  10. df = pd.read_excel(infile)
  11. table = pd.pivot_table(df, index=index_list,
  12. values=value_list,
  13. aggfunc=[np.sum, np.mean], fill_value=0)
  14. return table
  15.  
  16. def save_report(report, outfile):
  17. """
  18. Take a report and save it to a single Excel file
  19. """
  20. writer = pd.ExcelWriter(outfile)
  21. for manager in report.index.get_level_values(0).unique():
  22. temp_df = report.xs(manager, level=0)
  23. temp_df.to_excel(writer, manager)
  24. writer.save()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement