开发者

Extract Words from a file

开发者 https://www.devze.com 2023-02-10 07:37 出处:网络
I open a file using python to find whether a predefined set of words are present in the opened file or not. I took the predefined set of words in a list and opened the file that has to be tested. Now

I open a file using python to find whether a predefined set of words are present in the opened file or not. I took the predefined set of words in a list and opened the file that has to be tested. Now is there any method to extract words in py开发者_Go百科thon rather than lines. Thats makes my work lot easier.


import re

def get_words_from_string(s):
    return set(re.findall(re.compile('\w+'), s.lower()))

def get_words_from_file(fname):
    with open(fname, 'rb') as inf:
        return get_words_from_string(inf.read())

def all_words(needle, haystack):
    return set(needle).issubset(set(haystack))

def any_words(needle, haystack):
    return set(needle).intersection(set(haystack))

search_words = get_words_from_string("This is my test")
find_in = get_words_from_string("If this were my test, I is passing")

print any_words(search_words, find_in)

print all_words(search_words, find_in)

returns

set(['this', 'test', 'is', 'my'])
True


You can do a few things

  • Call file.readlines() and split the entire text on your desired delimiter if your text isn't large
  • Call read() and do it bytes at a time

Check out the pydocs for file - http://docs.python.org/release/2.5.2/lib/bltin-file-objects.html


This code will show what words are present in the file, given that the word exactly matches, and is not preceded or followed by punctuation or other characters, and is of the same case. With some minor adjustment, the code could be made more forgiving.

words = set(['hello', 'world', 'testing'])
f     = open('testfile.txt', 'rb')
data  = set(f.read().split())
print words.intersection(data)
0

精彩评论

暂无评论...
验证码 换一张
取 消