Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import os
- import pandas as pd
- # set path to files
- path = 'C:/Desktop/files'
- # site code list
- site_codes = ['COH','NCC','HCC','HCI','SIM','JPM','MCC','OSU','RPC','CNJ','SCC','UCC','MAR','UVA','USC','WCI']
- # empty list for dict keys
- df_names = []
- # list of filenames for dict values
- files = [os.path.join(path, f) for f in os.listdir(path)]
- # for each filename in files list, extract the site_code and append to df_names list
- for filename in files:
- if any(sub in filename for sub in site_codes):
- site = [sub_str for sub_str in site_codes if(sub_str in filename)]
- df_names.append(site)
- # create dict from lists
- file_dict = dict(zip(df_names, files))
Advertisement
Add Comment
Please, Sign In to add comment