Reputation: 103
I would like to text mine an excel file. First I must concatenate all rows into one large text file. Then, scan the text for words in a dictionary. If the word is found, count it as the dictionary key name. Finally return the list of counted words in a relational table [word, count]. I can count the words, but am unable to get the dictionary part to work. My question is:
tweaked code from the internet
import collections
import re
import matplotlib.pyplot as plt
import pandas as pd
#% matplotlib inline
#file = open('PrideAndPrejudice.txt', 'r')
#file = file.read()
''' Convert excel column/ rows into a string of words'''
#text_all = pd.read_excel('C:\Python_Projects\Rake\data_file.xlsx')
#df=pd.DataFrame(text_all)
#case_words= df['case_text']
#print(case_words)
#case_concat= case_words.str.cat(sep=' ')
#print (case_concat)
text_all = ("Billy was glad to see jack. Jack was estatic to play with Billy. Jack and Billy were lonely without eachother. Jack is tall and Billy is clever.")
''' done'''
import collections
import pandas as pd
import matplotlib.pyplot as plt
#% matplotlib inline
# Read input file, note the encoding is specified here
# It may be different in your text file
# Startwords
startwords = {'happy':'glad','sad': 'lonely','big': 'tall', 'smart': 'clever'}
#startwords = startwords.union(set(['happy','sad','big','smart']))
# Instantiate a dictionary, and for every word in the file,
# Add to the dictionary if it doesn't exist. If it does, increase the count.
wordcount = {}
# To eliminate duplicates, remember to split by punctuation, and use case demiliters.
for word in text_all.lower().split():
word = word.replace(".","")
word = word.replace(",","")
word = word.replace(":","")
word = word.replace("\"","")
word = word.replace("!","")
word = word.replace("“","")
word = word.replace("‘","")
word = word.replace("*","")
if word in startwords:
if word in wordcount:
wordcount[word] = 1
else:
wordcount[word] += 1
# Print most common word
n_print = int(input("How many most common words to print: "))
print("\nOK. The {} most common words are as follows\n".format(n_print))
word_counter = collections.Counter(wordcount)
for word, count in word_counter.most_common(n_print):
print(word, ": ", count)
# Close the file
#file.close()
# Create a data frame of the most common words
# Draw a bar chart
lst = word_counter.most_common(n_print)
df = pd.DataFrame(lst, columns = ['Word', 'Count'])
df.plot.bar(x='Word',y='Count')
Error: Empty 'DataFrame': no numeric data to plot
Expected output:
Upvotes: 2
Views: 336
Reputation: 18647
Here is a method that should work with the latest version of pandas
(0.25.3 at the time of writing):
# Setup
df = pd.DataFrame({'case_text': ["Billy was glad to see jack. Jack was estatic to play with Billy. Jack and Billy were lonely without eachother. Jack is tall and Billy is clever."]})
startwords = {"happy":["glad","estatic"],
"sad": ["depressed", "lonely"],
"big": ["tall", "fat"],
"smart": ["clever", "bright"]}
# First you need to rearrange your startwords dict
startwords_map = {w: k for k, v in startwords.items() for w in v}
(df['case_text'].str.lower() # casts to lower case
.str.replace('[.,\*!?:]', '') # removes punctuation and special characters
.str.split() # splits the text on whitespace
.explode() # expands into a single pandas.Series of words
.map(startwords_map) # maps the words to the startwords
.value_counts() # counts word occurances
.to_dict()) # outputs to dict
[out]
{'happy': 2, 'big': 1, 'smart': 1, 'sad': 1}
Upvotes: 3
Reputation: 1411
if word in startwords:
if word in wordcount:
wordcount[word] = 1
else:
wordcount[word] += 1
This part seems problematic, it checks if word
in startwords
and then further check in wordcount
, if it's in the wordcount
, it should increase the word count by your logic. So I believe you have to switch the execution.
if word in wordcount:
//in dict, count++
wordcount[word] += 1
else:
// first time, set to 1
wordcount[word] = 1
Upvotes: 2