m8_

dataframe to dict

m8_
Apr 1st, 2019
172
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.68 KB | None | 0 0
  1. import os
  2. import pandas as pd
  3.  
  4. # set path to files
  5. path = 'C:/Desktop/files'
  6.  
  7. # site code list
  8. site_codes = ['COH','NCC','HCC','HCI','SIM','JPM','MCC','OSU','RPC','CNJ','SCC','UCC','MAR','UVA','USC','WCI']
  9.  
  10. # empty list for dict keys
  11. df_names = []
  12.  
  13. # list of filenames for dict values
  14. files = [os.path.join(path, f) for f in os.listdir(path)]
  15.  
  16. # for each filename in files list, extract the site_code and append to df_names list
  17. for filename in files:
  18.     if any(sub in filename for sub in site_codes):
  19.         site = [sub_str for sub_str in site_codes if(sub_str in filename)]
  20.         df_names.append(site)
  21.  
  22. # create dict from lists
  23. file_dict = dict(zip(df_names, files))
Advertisement
Add Comment
Please, Sign In to add comment