这篇博客提供能够输出小说人物关系的完整python代码。
代码的原作者是Qingyu Mao,十分感谢!!
以下是Mao的github链接和博客教程链接:
@source: https://github.com/maoqyhz/TextCharactervVisualization
@tutorial: https://www.cnblogs.com/Sinte-Beuve/p/7679392.html
获得小说中人物关系数据和对关系进行可视化的具体理论思路、操作方法、数据准备请参考Qingyu Mao的博客,ta已经讲得非常清楚详细了!
Mao的代码可能存在输出只有表头的空csv的问题,我对代码稍作修改后得到了正常输出。我也添加了两个简单的checkpoint,供有需要的朋友检查代码不能运行的原因。修正后的所有代码如下:
"""
Created on 2017/10/15 19:24
Modified on 2022/05/03 14:06
@author: Qingyu Mao
@source: https://github.com/maoqyhz/TextCharactervVisualization
@tutorial: https://www.cnblogs.com/Sinte-Beuve/p/7679392.html
@co-author: 农魔饼饼
"""
import jieba
import codecs
from collections import Counter
from collections import defaultdict
from __future__ import print_function
TEXT_PATH = './Desktop/text.tx