keywords_year.py 文件源码

def write_keywords(topK):

    data = utils.read_file(source_file)
    dic = get_keywords_year(data)

    with open(keywords_year_file, 'w', newline='') as csvfile:
        writer = csv.writer(csvfile)
        for year in year_range:
            row = [ year ]
            count = 0
            for k, v in dic[year]:
                row.append(k + ':' + str(v))
                count = count + 1
                if count == topK:
                    break
            writer.writerow(row)