Function frequency_counter

LSA/LSA.py:58–74

INPUT: text - (list) 文本列表 words - (list) 单词列表 OUTPUT: X - (array) 单词-文本矩阵

(text, words)

#定义构建单词-文本矩阵的函数，这里矩阵的每一项表示单词在文本中的出现频次，也可以用TF-IDF来表示
def frequency_counter(text, words):
    '''
    INPUT:
    text - (list) 文本列表
    words - (list) 单词列表
    
    OUTPUT:
    X - (array) 单词-文本矩阵
    
    '''
    X = np.zeros((len(words), len(text)))  #定义m*n的矩阵，其中m为单词列表中的单词个数，n为文本个数
    for i in range(len(text)):
        t = text[i]  #读取文本列表中的第i条文本
        for w in t:
            ind = words.index(w)  #取出第i条文本中的第t个单词在单词列表中的索引
            X[ind][i] += 1  #对应位置的单词出现频次加一
    return X


#定义潜在语义分析函数