In [1]:

    

import os
import pandas as pd


    

In [2]:

    

class polar_classifier:
    def __init__(self, path_to_dict = '/home/da/emotion_dict.xlsx'):
        try:
            self.emotion_dict = pd.read_excel('/home/da/emotion_dict.xlsx')
        except FileNotFoundError as e:
            print('当前词典路径错误:%s' % (e))
        self.polar_dict = self.emotion_dict.loc[:, ['词语', '极性']]
        self.polar_dict.rename(columns={'词语':'word', '极性':'polar'}, inplace = True)
        self.polar_dict = self.polar_dict.drop_duplicates('word')
        self.pos_dict = set(self.polar_dict.loc[self.polar_dict.polar==1, 'word'])
        self.neg_dict = set(self.polar_dict.loc[self.polar_dict.polar==2, 'word'])
        
    def classify(self, word):
        if word in self.pos_dict:
            return 1
        elif word in self.neg_dict:
            return-1
        else:
            return 0
        
    @property
    def show_pos(self):
        print(self.pos_dict)
        
    @property
    def show_neg(self):
        print(self.neg_dict)


    

In [3]:

    

a = polar_classifier()
a.classify('开心')


    

    
Out[3]:





1