sorted_avg_chi = sorted(dict_avg_chi.items(), key=lambda(k, v): (v, k), reverse=True)
# Sort Max CHI SCore dictionary by value
sorted_max_chi = sorted(dict_max_chi.items(), key=lambda(k, v): (v, k), reverse=True)
list_avg_terms = []
list_max_terms = []
x = 1
for key, val in sorted_avg_chi:
if x > limit:
break
else:
list_avg_terms.append(key)
x += 1
y = 1
for key, val in sorted_max_chi:
if y > limit:
break
else:
list_max_terms.append(key)
y += 1
for cate in range(1, 10):
generateGexf(list_avg_terms, cate, 'avg')
generateGexf(list_max_terms, cate, 'max')
# Close connection
mysql.disconnect()
