-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
52 lines (39 loc) · 1.87 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import string
from collections import Counter
import matplotlib.pyplot as plt
txt=open('data.txt',encoding='utf-8').read()
lw=txt.lower()
cleaned_txt = lw.translate(str.maketrans('','',string.punctuation))
#print(cleaned_txt)
t_w = cleaned_txt.split()
#print(t_w)
stop_words = ["i", "me", "my", "myself", "we", "our", "ours", "ourselves", "you", "your", "yours", "yourself",
"yourselves", "he", "him", "his", "himself", "she", "her", "hers", "herself", "it", "its", "itself",
"they", "them", "their", "theirs", "themselves", "what", "which", "who", "whom", "this", "that", "these",
"those", "am", "is", "are", "was", "were", "be", "been", "being", "have", "has", "had", "having", "do",
"does", "did", "doing", "a", "an", "the", "and", "but", "if", "or", "because", "as", "until", "while",
"of", "at", "by", "for", "with", "about", "against", "between", "into", "through", "during", "before",
"after", "above", "below", "to", "from", "up", "down", "in", "out", "on", "off", "over", "under", "again",
"further", "then", "once", "here", "there", "when", "where", "why", "how", "all", "any", "both", "each",
"few", "more", "most", "other", "some", "such", "no", "nor", "not", "only", "own", "same", "so", "than",
"too", "very", "s", "t", "can", "will", "just", "don", "should", "now"]
final_words = []
for word in t_w:
if word not in stop_words:
final_words.append(word)
#print(final_words)
em_list = []
with open('emotion.txt','r') as emo:
for l in emo:
c_l = l.replace("\n","").replace("'",'').strip()
w,em = c_l.split(":")
if w in final_words:
em_list.append(em)
print(em_list)
c= Counter(em_list)
print(c)
fig, ax1 = plt.subplots()
ax1.bar(c.keys(), c.values())
fig.autofmt_xdate()
plt.savefig('result.png')
plt.show()