三国演义词云的python代码只留人名
时间: 2023-10-12 20:27:57 浏览: 185
好的,以下是一个简单的Python代码,用于生成三国演义中人名的词云。首先,我们需要使用jieba库将文本中的所有词语分割开来,再使用wordcloud库生成词云图。
```python
import jieba
from wordcloud import WordCloud
import matplotlib.pyplot as plt
# 读取文本
with open('three_kingdoms.txt', 'r', encoding='utf-8') as f:
text = f.read()
# 使用jieba分词
words = jieba.cut(text)
# 过滤出人名
names = []
for word in words:
if len(word) == 2 and word[0] != ' ':
names.append(word)
# 生成词云图
wc = WordCloud(background_color='white', width=800, height=600, font_path='msyh.ttc')
wc.generate(' '.join(names))
plt.imshow(wc)
plt.axis('off')
plt.show()
```
在代码中,我们首先读取了三国演义的文本文件,然后使用jieba库进行分词。接着,我们过滤出长度为2且第一个字符不为空格的词语,这些词语很有可能是人名。最后,我们使用wordcloud库生成词云图,并将人名列表作为输入。
注意,这只是一个简单的示例代码,可能无法完全准确地过滤出所有人名。如果需要更高的准确度,需要使用更复杂的算法和模型。
阅读全文