Python day2--生成词云

先看下效果：

代码如下：

自己准备一个txt文件，写入单词即可

# import
from wordcloud import WordCloud
import matplotlib.pyplot as plt
import csv

# 打开文件
file = open(r"C:\Users\lsy\Desktop\1234.txt")

# reader object 
reader = csv.reader(file)

# contents of reader object is stored .
# data is stored in list of list  format.
reader_contents = list(reader)

text = ""

# iterating through list of rows
for row in reader_contents :

 #  iterating through words in the row
 for word in row :

  # concatenate the words
  text = text + " " + word

# show only 10 words in the wordcloud .
wordcloud = WordCloud(width=480,  height=480, max_words=10).generate(text)

# plot the WordCloud image
plt.figure()
plt.imshow(wordcloud,  interpolation="bilinear")
plt.axis("off")
plt.margins(x=0, y=0)
plt.show()