UCINET网络图绘制的Python代码

Question

我使用 beautifulsoup 抓取了一些数据，并保存为 .txt 文件。数据是来自 IMDB.com 的电影评论 我找到了一个很好的单词计数python代码，所以我可以制作一个词频excel表。但是，我无法仅使用频率表绘制图表。

我想使用UCINET绘制语义网络图（节点大小应该基于中介中心性。）

我的问题是如何将文本文件转换为邻接矩阵数据来绘制UCINET图。 像这样http://www.umasocialmedia.com/socialnetworks/wp-content/uploads/2012/09/senatorsxsenators1.png 我想用审稿人的话来画网络图。

（计算两个词在同一个句子中出现的频率，当它们匹配行和列行时）

或者。你能告诉我如何在 Python 代码中绘制网络图（使用中介中心性）吗？

Answer 1

制作一个 2D 20x20 数组，遍历每个输入字符串，并使用该字符串更新您的矩阵：

adjacency_matrix = [[0 for _ in range(20)] for _ in range(20)]


def get_lines(filename):
    """Returns the lines in the file"""
    with open(filename, 'r') as fp:
        return fp.readlines()


def update_matrix(matrix, mapping, string):
    """Update the given adjacency matrix using the given string."""                                        
    words = [_ for _ in re.split("\s+", string) if _ in mapping.keys()]            
    for word_1 in words:                                                           
        for word_2 in words:                                                       
            matrix[mapping[word_1]][mapping[word_2]] += 1


if __name__ == "__main__":
    words_in_matrix = ["X-men", "awesome", "good", "bad", ... 16 more ...]
    mapping = {word: index for index, word in enumerate(words_in_matrix)}

    for line in get_lines("ibdb.txt"):
        update_matrix(adjacency_matrix, mapping, line)
    print(adjacency_matrix)

类似于update_matrix 的函数可能有用，matrix 作为邻接矩阵，mapping 是单词到邻接矩阵中索引的映射，string 是您的示例评论。

您需要根据自己的需要对其进行修改。输入可能有句点或其他干扰字符，需要去除。