parent
a13b592466
commit
3593705de6
@ -0,0 +1,39 @@
|
|||||||
|
import jieba
|
||||||
|
from matplotlib import pyplot as plt
|
||||||
|
from wordcloud import WordCloud
|
||||||
|
from PIL import Image
|
||||||
|
import numpy as np
|
||||||
|
import sqlite3
|
||||||
|
|
||||||
|
def makeWordcloud(image_name):
|
||||||
|
con = sqlite3.connect('data.db')
|
||||||
|
cur = con.cursor()
|
||||||
|
sql = 'select info from movie250 '
|
||||||
|
data = cur.execute(sql)
|
||||||
|
text = ""
|
||||||
|
for item in data:
|
||||||
|
text = text + item[0]
|
||||||
|
cur.close()
|
||||||
|
con.close()
|
||||||
|
cut = jieba.cut(text)
|
||||||
|
string = ' '.join(cut)
|
||||||
|
img = Image.open(image_name)
|
||||||
|
img_array = np.array(img)
|
||||||
|
wc = WordCloud(
|
||||||
|
# background_color = 'white',
|
||||||
|
# mask = img_array,
|
||||||
|
# font_path = 'simsun.ttc'
|
||||||
|
font_path="msyhbd.ttc",
|
||||||
|
mask=img_array,
|
||||||
|
width=1500,
|
||||||
|
height=1000,
|
||||||
|
background_color="white",
|
||||||
|
min_font_size=5,
|
||||||
|
max_font_size=120,
|
||||||
|
max_words=600
|
||||||
|
)
|
||||||
|
wc.generate_from_text(string)
|
||||||
|
fig = plt.figure(1)
|
||||||
|
plt.imshow(wc)
|
||||||
|
plt.axis('off')
|
||||||
|
plt.savefig('output.jpg', dpi=1600)
|
Loading…
Reference in new issue