# -*- coding: utf-8 -*-
from collections import defaultdict
import math
import operator

"""
函數說明:創(chuàng)建數據樣本
Returns:
    dataset - 實驗樣本切分的詞條
    classVec - 類別標簽向量
"""


def loadDataSet():
    dataset = [['my', 'dog', 'has', 'flea', 'problems', 'help', 'please'],  # 切分的詞條
               ['maybe', 'not', 'take', 'him', 'to', 'dog', 'park', 'stupid'],
               ['my', 'dalmation', 'is', 'so', 'cute', 'I', 'love', 'my'],
               ['stop', 'posting', 'stupid', 'worthless', 'garbage'],
               ['mr', 'licks', 'ate', 'my', 'steak', 'how', 'to', 'stop', 'him'],
               ['quit', 'buying', 'worthless', 'dog', 'food', 'stupid']]
    classVec = [0, 1, 0, 1, 0, 1]  # 類別標簽向量，1代表好，0代表不好
    return dataset, classVec


"""
函數說明：特征選擇TF-IDF算法
Parameters:
     list_words:詞列表
Returns:
     dict_feature_select:特征選擇詞字典
"""
#dataset:文件夾，word_list:某一個文件，word某個詞

def feature_select(dataset):
    # 總詞頻統計
    doc_frequency = defaultdict(int) #記錄每個詞出現的次數，可以把它理解成一個可變長度的list，只要你索引它，它就自動擴列
    for file in dataset:
        for word in file:
            doc_frequency[word] += 1
    # 計算每個詞的TF值
    word_tf = {}  # 存儲沒個詞的tf值
    for i in doc_frequency:
        word_tf[i] = doc_frequency[i] / sum(doc_frequency.values()) #sum(doc.frequency.values)

    # 計算每個詞的IDF值
    doc_num = len(dataset)
    word_idf = {}  # 存儲每個詞的idf值
    word_doc = defaultdict(int)  # 存儲包含該詞的文檔數
    for word in doc_frequency:
        for file in dataset:
            if word in file:
                word_doc[word] += 1
    #word_doc和doc_frequency的區(qū)別是word_doc存儲的是包含這個詞的文檔數，即如果一個文檔里有重復出現一個詞則word_doc < doc_frequency
    for word in doc_frequency:
        word_idf[word] = math.log(doc_num / (word_doc[word] + 1))

    # 計算每個詞的TF*IDF的值
    word_tf_idf = {}
    for word in doc_frequency:
        word_tf_idf[word] = word_tf[word] * word_idf[word]

    # 對字典按值由大到小排序
    dict_feature_select = sorted(word_tf_idf.items(), key=operator.itemgetter(1), reverse=True)
    return dict_feature_select


if __name__ == '__main__':
    data_list, label_list = loadDataSet()  # 加載數據
    features = feature_select(data_list)  # 所有詞的TF-IDF值
    print(features)

from nltk.text import TextCollection
from nltk.tokenize import word_tokenize
 
#首先，構建語料庫corpus
sents=['this is sentence one','this is sentence two','this is sentence three']
sents=[word_tokenize(sent) for sent in sents] #對每個句子進行分詞
print(sents)  #輸出分詞后的結果
corpus=TextCollection(sents)  #構建語料庫
print(corpus)  #輸出語料庫
 
#計算語料庫中"one"的tf值
tf=corpus.tf('one',corpus)    # 1/12
print(tf)
 
#計算語料庫中"one"的idf值
idf=corpus.idf('one')      #log(3/1)
print(idf)
 
#計算語料庫中"one"的tf-idf值
tf_idf=corpus.tf_idf('one',corpus)
print(tf_idf)

import sklearn
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.feature_extraction.text import TfidfTransformer

x_train = ['TF-IDF 主要 思想 是', '算法 一個 重要 特點 可以 脫離 語料庫 背景',
           '如果 一個 網頁 被 很多 其他 網頁 鏈接 說明 網頁 重要']
x_test = ['原始 文本 進行 標記', '主要 思想']

# 該類會將文本中的詞語轉換為詞頻矩陣，矩陣元素a[i][j] 表示j詞在i類文本下的詞頻
vectorizer = CountVectorizer(max_features=10) #列數為10
# 該類會統計每個詞語的tf-idf權值
tf_idf_transformer = TfidfTransformer()
# 將文本轉為詞頻矩陣并計算tf-idf
tf_idf = tf_idf_transformer.fit_transform(vectorizer.fit_transform(x_train))
# 將tf-idf矩陣抽取出來，元素a[i][j]表示j詞在i類文本中的tf-idf權重
x_train_weight = tf_idf.toarray()

# 對測試集進行tf-idf權重計算
tf_idf = tf_idf_transformer.transform(vectorizer.transform(x_test))
x_test_weight = tf_idf.toarray()  # 測試集TF-IDF權重矩陣

print('vectorizer.fit_transform(x_train) : ')
print(vectorizer.fit_transform(x_train))
print('輸出x_train文本向量：')
print(x_train_weight)
print('輸出x_test文本向量：')
print(x_test_weight)

import jieba.analyse
 
text='關鍵詞是能夠表達文檔中心內容的詞語，常用于計算機系統標引論文內容特征、
信息檢索、系統匯集以供讀者檢閱。關鍵詞提取是文本挖掘領域的一個分支，是文本檢索、
文檔比較、摘要生成、文檔分類和聚類等文本挖掘研究的基礎性工作'
 
keywords=jieba.analyse.extract_tags(text, topK=5, withWeight=False, allowPOS=())
print(keywords)

jieba.analyse.extract_tags(sentence, topK=20, withWeight=False, allowPOS=())