从CSV文件中的列创建Python中的文字云

时间:2017-09-13 16:29:01

标签: python csv word-cloud

我正在尝试使用python从csv文件中的列创建一个词云。我已经安装了wordcloud,PIL,OS和Numpy,但我不确定如何阅读这一列来创建单词云。有人能协助吗?谢谢!

1 个答案:

答案 0 :(得分:1)

以下是word_cloud site的修改示例。

我在第一列中有一个'words.csv文件,其中包含单词(在下面的图片中被选中并显示在下面)以及下一页中的说明。

我们在csv文件(字词)上阅读第一栏,然后浏览word_cloud

import csv
from wordcloud import WordCloud


#read first column of csv file to string of words seperated
#by tab

your_list = []
with open('words.csv', 'rb') as f:
    reader = csv.reader(f)
    your_list = '\t'.join([i[0] for i in reader])


# Generate a word cloud image
wordcloud = WordCloud().generate(your_list)

# Display the generated image:
# the matplotlib way:
import matplotlib.pyplot as plt
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis("off")

# lower max_font_size
wordcloud = WordCloud(max_font_size=40).generate(your_list)
plt.figure()
plt.imshow(wordcloud, interpolation="bilinear")
plt.axis("off")
plt.show()

# The pil way (if you don't have matplotlib)
# image = wordcloud.to_image()
# image.show()

图片
enter image description here