Files
undergrad-uh401/excel_tools.py
2025-12-06 03:59:17 +00:00

31 lines
746 B
Python

import pandas as pd
# Keys
key_all = "All R&D expenditures"
key_federal = "Federal government"
key_state = "State and local government"
key_inst = "Institution funds"
key_business = "Business"
key_nonprofit = "Nonprofit organizations"
key_other = "All other sources"
keys = [key_federal, key_state, key_inst, key_business, key_nonprofit, key_other]
percent_suffix = " Percent"
def read_excel(file):
# Read excel
df = pd.read_excel(file, header=3)
# Remove fluff
df = df[:-4][[field for field in df[:-4] if 'Unnamed' not in field]]
# Add percent
for key in keys:
key_percent = key + percent_suffix
df[key_percent] = df[key] / df[key_all] * 100
return df
def print_data(data):
print(data)