def calc_freq(date_from,date_to,window_size,wanted_list,n,search=[]): groups,wanted_list = group(wanted_list) # list of tuples of dates all_dates = Windows.get_dates(date_from,date_to,365) # these are not windows, smaller problems # for dates and frequencies x = [] y = np.empty((len(wanted_list), 0)).tolist() stevec = 0 print('grem v zanke') for tup in all_dates: print('leto') windows, time = Windows.get_windows(tup[0], tup[1], window_size,search) print('windows') x += time if n==1: grams = count_words(wanted_list,windows) else: grams, words_per_window = N_grams.list_of_ngrams(windows, n) print('ngrami narjeni') for count, i in enumerate(grams): #words = words_per_window[count] #to calculate frequency #besede.append(words) words = besede[stevec] stevec+=1 for j,wanted in enumerate(wanted_list): frequency = 0 if wanted in i: frequency = i[wanted] / words y[j].append(frequency) print('ngrami presteti') #print(besede) print('plotam') #print(x,y,wanted_list,groups) plot(x,y,wanted_list,groups) return x,y