Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import argparse
- import pandas as pd
- import numpy as np
- def create_pivot(infile, index_list=["Manager", "Rep", "Product"],
- value_list=["Price", "Quantity"]):
- """
- Read in the Excel file, create a pivot table and return it as a DataFrame
- """
- df = pd.read_excel(infile)
- table = pd.pivot_table(df, index=index_list,
- values=value_list,
- aggfunc=[np.sum, np.mean], fill_value=0)
- return table
- def save_report(report, outfile):
- """
- Take a report and save it to a single Excel file
- """
- writer = pd.ExcelWriter(outfile)
- for manager in report.index.get_level_values(0).unique():
- temp_df = report.xs(manager, level=0)
- temp_df.to_excel(writer, manager)
- writer.save()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement