DWizard commited on
Commit
7def859
·
1 Parent(s): 9ed44bb

improve dict_util

Browse files

Former-commit-id: 081d4404a6e82d5dabf1f34654a3a8560c9cd98e

Files changed (1) hide show
  1. dict_util.py +6 -10
dict_util.py CHANGED
@@ -7,11 +7,12 @@ import pickle
7
  # 1_2_3, 1 is action, 2 is supply object, 3 is accept object
8
  def update_dict_csv(term_dict, f):
9
  for rows in csv.reader(f):
10
- if rows[0] in term_dict:
11
- if rows[1] not in term_dict[rows[0]]:
12
- term_dict[rows[0]] = term_dict[rows[0]]+[rows[1]]
 
13
  else:
14
- term_dict[rows[0]]=[rows[1]]
15
  pass
16
 
17
  def export_dict_csv(term_dict, f):
@@ -24,12 +25,7 @@ def save_dict_pickle(term_dict, f):
24
 
25
  def update_csv_pickle(pickle_f, csv_f):
26
  term_dict = pickle.load(pickle_f)
27
- for rows in csv.reader(csv_f):
28
- if rows[0] in term_dict:
29
- if rows[1] not in term_dict[rows[0]]:
30
- term_dict[rows[0]] = term_dict[rows[0]]+[rows[1]]
31
- else:
32
- term_dict[rows[0]]=[rows[1]]
33
  #save to pickle file, highest protocal to get better performance
34
  pickle.dump(term_dict, pickle_f, pickle.HIGHEST_PROTOCOL)
35
 
 
7
  # 1_2_3, 1 is action, 2 is supply object, 3 is accept object
8
  def update_dict_csv(term_dict, f):
9
  for rows in csv.reader(f):
10
+ word = rows[0].lower()
11
+ if word in term_dict:
12
+ if rows[1] not in term_dict[word]:
13
+ term_dict[word] = term_dict[word]+[rows[1]]
14
  else:
15
+ term_dict[word]=[rows[1]]
16
  pass
17
 
18
  def export_dict_csv(term_dict, f):
 
25
 
26
  def update_csv_pickle(pickle_f, csv_f):
27
  term_dict = pickle.load(pickle_f)
28
+ update_dict_csv(term_dict, csv_f)
 
 
 
 
 
29
  #save to pickle file, highest protocal to get better performance
30
  pickle.dump(term_dict, pickle_f, pickle.HIGHEST_PROTOCOL)
31