×

Welcome to TagMyCode

Please login or create account to add a snippet.
 

extract_sheets_from_excel

0
0
 
0
Language: Python
Posted by: Chelsea Trotter
Added: Apr 5, 2022 5:18 PM
Views: 405
Tags: readexcel
  1. def extract_sheets_from_excel(file, pattern):
  2.     df = pd.DataFrame()
  3.     # num_sheets = len(pd.ExcelFile(file).sheet_names)
  4.     sheets = [sheet for sheet in pd.ExcelFile(file).sheet_names if pattern in sheet]
  5.     for sheet in sheets:
  6.         print(f"Processing data sheet {sheet}")
  7.         if pattern == "samples":
  8.             colrange = "A:R"
  9.         if pattern == "files":
  10.             colrange = "A:N"
  11.         excel_data = pd.read_excel(file, sheet_name=sheet, skiprows=2, usecols=colrange)
  12.         # check_if_pattern_col_is_na(excel_data, "Unnamed")
  13.         excel_data = unify_col_name(excel_data)
  14.         df = pd.concat([df, excel_data], axis=0)
  15.     return df

Write a comment

To post comments, please log in or register.