用Python计算文件中的单词

Question

I'm completely new to Python, but to my own surprise I've produced this working piece of code:

if __name__ == "__main__":
with open("wordlist.txt") as infile:
    for line in infile:
        print(line)    



with open ("cv000_29416.txt", "r") as myfile:
   data=myfile.read().replace('\n', '')
print (data.count("bad"))      

The point is, that I want to count the words from wordlist.txt in cv000_29416.txt.

(So wordlist.txt contains for example twenty words like 'bad', 'good' etcetera, and cv000_29416.txt is a long text, and I want to count how many time 'bad', 'good' etcetera occure in cv000_29416.txt)

Can I insert that somewhere in the seconds piece of code?

Thank you! and sorry for bad English

Answer 1

# create a collection of the words that want to count
with open('wordlist.txt') as infile:
    counts = {}
    for line in infile:
        for word in line.split():
            counts[word] = 0

# increment the count of the words that you really care about
with open("cv000_29416.txt") as infile:
    for line in infile:
        for word in line.split():
            if word in counts:
                counts[word] += 1

for word,count in counts.items():
    print(word, "appeared", count, "times")

Answer 2

use a collections.Counter dict to count all the words:

from collections import Counter
with open ("cv000_29416.txt", "r") as myfile:
   data = Counter(myfile.read().split())
print (data["bad"])   

To put it together, presuming each word is on a separate line in wordlist.txt:

from collections import Counter
with open ("cv000_29416.txt", "r") as myfile,open("wordlist.txt") as infile:
    data = Counter(myfile.read().split())
    for line in infile:
        print(data.get(line.rstrip(),0))