main.py 文件源码

def generate_img(data):
    mask_img = imread('./heart-mask.jpg')
    wordcloud = WordCloud(
        font_path='/Library/Fonts/Songti.ttc',
        background_color='white',
        mask=mask_img
    ).generate(data)
    plt.imshow(wordcloud)
    plt.axis('off')
    # plt.show()
    plt.savefig('./heart.jpg', dpi=600)