Skip to content

Instantly share code, notes, and snippets.

@thmcmahon
Created October 27, 2017 01:01
Show Gist options
  • Save thmcmahon/142ecb8dc5d58f818b5143b683e8d976 to your computer and use it in GitHub Desktop.
Save thmcmahon/142ecb8dc5d58f818b5143b683e8d976 to your computer and use it in GitHub Desktop.
#!/usr/local/bin/python3
import pandas as pd
import glob
import os
import sys
input_path = sys.argv[1]
output_file = sys.argv[2]
all_workbooks = glob.glob(os.path.join(input_path,'*.xls*'))
df_names = []
for i in all_workbooks:
df_names.append(os.path.basename(i)[:-5])
data_frames = []
for workbook in all_workbooks:
all_worksheets = pd.read_excel(workbook, sheetname=0, index_col=None)
data_frames.append(all_worksheets)
all_data_concatenated = pd.concat(data_frames, keys=df_names)
all_data_concatenated.reset_index(level=0, inplace=True)
all_data_concatenated.rename(columns={'level_0': 'Shadow'}, inplace=True)
writer = pd.ExcelWriter(output_file)
all_data_concatenated.to_excel(writer, sheet_name='all_data_all_workbooks', index=False)
writer.save()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment