def emotCount(df):
emotString = ''
for i in range(0,len(df)):
emotString = emotString + ' ' + str(df[i]['SIT'].str.cat())
cleanEmot = re.sub(r"[^\w\d'\s]+",' ', emotString.lower())
cleanEmot = cleanEmot.split()
emotCount = dictCount(cleanEmot)
return(emotCount)
allStories = textClean()
for k in range(0,len(allStories)-1):
story = allStories[k]
happy = [line.rstrip('\n') for line in open('Stories/emotions/happy.txt')]
neg = [line.rstrip('\n') for line in open('Stories/emotions/negative.txt')]
#allEmotion = happy+neg
totHap = []
totNeg = []
totSplit = 26
split = len(story) / totSplit
for i in range(0,totSplit-1):
textChunk = dictCount(story[split*i:split*(i+1)])
hCount = 0.0
nCount = 0.0
for j in range(0, len(happy)):
hCount += textChunk[happy[j]]
for j in range(0,len(neg)):
nCount += textChunk[neg[j]]
totHap.append(hCount)
totNeg.append(nCount)
print ("Number of Happy Words: " + str(totHap))
print ("Number of Negative Words: " + str(totNeg))
plt.plot(totHap)
plt.plot(totNeg)
