Python3紅樓夢人名出現(xiàn)次數(shù)統(tǒng)計分析

import jieba
from collections import Counter
import matplotlib.pyplot as plt
import numpy as np


class HlmNameCount():
    # 此函數(shù)用于繪制條形圖
    def showNameBar(self,name_list_sort,name_list_count):
        # x代表條形數(shù)量
        x = np.arange(len(name_list_sort))
        # 處理中文亂碼
        plt.rcParams['font.sans-serif'] = ['SimHei']
        # 繪制條形圖，bars相當于句柄
        bars = plt.bar(x,name_list_count)
        # 給各條形打上標簽
        plt.xticks(x,name_list_sort)
        # 顯示各條形具體數(shù)量
        i = 0
        for bar in bars:
            plt.text((bar.get_x() + bar.get_width() / 2), bar.get_height(), '%d' % name_list_count[i], ha='center', va='bottom')
            i += 1
        # 顯示圖形
        plt.show()

    # 此函數(shù)用于繪制餅狀圖
    def showNamePie(self, name_list_sort, name_list_fracs):
        # 處理中文亂碼
        plt.rcParams['font.sans-serif'] = ['SimHei']
        # 繪制餅狀圖
        plt.pie(name_list_fracs, labels=name_list_sort, autopct='%1.2f%%', shadow=True)
        # 顯示圖形
        plt.show()

    def getNameTimesSort(self,name_list,txt_path):
        # 將所有人名臨時添加到jieba所用字典，以使jieba能識別所有人名
        for k in name_list:
            jieba.add_word(k)
        # 打開并讀取txt文件
        file_obj = open(txt_path, 'rb').read()
        # jieba分詞
        jieba_cut = jieba.cut(file_obj)
        # Counter重新組裝以方便讀取
        book_counter = Counter(jieba_cut)
        # 人名列表，因為要處理鳳姐所以不直接用name_list
        name_dict ={}
        # 人名出現(xiàn)的總次數(shù)，用于后邊計算百分比
        name_total_count = 0
        for k in name_list:
            if k == '熙鳳':
                # 將熙鳳出現(xiàn)的次數(shù)合并到鳳姐
                name_dict['鳳姐'] += book_counter[k]
            else:
                name_dict[k] = book_counter[k]
            name_total_count += book_counter[k]
        # Counter重新組裝以使用most_common排序
        name_counter = Counter(name_dict)
        # 按出現(xiàn)次數(shù)排序后的人名列表
        name_list_sort = []
        # 按出現(xiàn)次數(shù)排序后的人名百分比列表
        name_list_fracs = []
        # 按出現(xiàn)次數(shù)排序后的人名次數(shù)列表
        name_list_count = []
        for k,v in name_counter.most_common():
            name_list_sort.append(k)
            name_list_fracs.append(round(v/name_total_count,2)*100)
            name_list_count.append(v)
            # print(k+':'+str(v))
        # 繪制條形圖
        self.showNameBar(name_list_sort, name_list_count)
        # 繪制餅狀圖
        self.showNamePie(name_list_sort,name_list_fracs)
        

if __name__ == '__main__':
    # 參與統(tǒng)計的人名列表，可修改成自己想要的列表
    name_list = ['寶玉', '黛玉', '寶釵', '元春', '探春', '湘云', '妙玉', '迎春', '惜春', '鳳姐', '熙鳳', '巧姐', '李紈', '可卿', '賈母', '賈珍', '賈蓉', '賈赦', '賈政', '王夫人', '賈璉', '薛蟠', '香菱', '寶琴', '襲人', '晴雯', '平兒', '紫鵑', '鶯兒']
    # 紅樓夢txt文件所在路徑，修改成自己文件所在路徑
    txt_path = 'F:/PycharmProjects/tutorial/hlm.txt'
    hnc = HlmNameCount()
    hnc.getNameTimesSort(name_list,txt_path)