# -*- coding:utf-8 -*- """ author: Ziyu Chen """ from poem import Poem if __name__ == '__main__': poem = Poem('poetry_simplified.txt') poem.set_stopwords('stopwords.txt') poem.process() poem_set = poem.return_poem_set() poem.generate_word_cloud('poetry_simplified.txt', 'jiubei.jpg', 'jiubei_cw.jpg', userpic_as_backgroud=False) poem.generate_wordcloud_for_poet('李白', 'libai.jpg', 'libai_cw.jpg') poem.generate_wordcloud_for_poet('杜甫', 'dufu.jpg', 'dufu_cw.jpg') poem.draw_top_produced_poet() poem.draw_top_poem_kind() # poem.word_statistics(15)