如何将枚举中的值与关键字匹配？

Question

我想编写一个上下文中的关键字脚本，其中我首先读取一个文本文件作为枚举列表，然后 return 一个给定的关键字和接下来的五个词。

我看到类似的问题被问到 C# 并且我在 Python 中找到了 enum 模块的解决方案，但我希望有一个仅使用 enumerate() 函数的解决方案.

这是我目前得到的：

# Find keywords in context

import string

# open input txt file from local path

with open('C:\Users\somefile.txt', 'r', encoding='utf-8', errors='ignore') as f: # open file
    data1=f.read() # read content of file as string
    data2=data1.translate(str.maketrans('', '', string.punctuation)).lower() # remove punctuation
    data3=" ".join(data2.split()) # remove additional whitespace from text
    indata=list(data3.split()) # convert string to list
    print(indata[:4])
    
searchterms=["text", "book", "history"]
 
def wordsafter(keyword, source):
    for i, val in enumerate(source):
        if val == keyword: # cannot access the enumeration value here
            return str(source[i+5]) # intend to show searchterm and subsequent five words
        else:
            continue

for s in searchterms: # iterate through searchterms
    print(s)
    wordsafter(s, indata)
    
print("done")

我希望我可以像在这里所做的那样简单地访问枚举的值，但情况似乎并非如此。

Answer 1

感谢@jasonharper，您改进的代码：

import string


def wordsafter(keyword, source):
    for i, val in enumerate(source):
        if val == keyword:
            return ' '.join(source[i:i + 5])  # intend to show searchterm and subsequent five words

# wordsafter() for all instances
def wordsafter(keyword, source):
     instances = []
     for i, val in enumerate(source):
        if val == keyword:
            instances.append(' '.join(source[i:i + 5]))
     return instances

# open input txt file from local path
with open('README.md', 'r', encoding='utf-8', errors='ignore') as f:  # open file
    data1 = f.read()  # read content of file as string
    data2 = data1.translate(str.maketrans('', '', string.punctuation)).lower()  # remove punctuation
    data3 = " ".join(data2.split())  # remove additional whitespace from text
    indata = list(data3.split())  # convert string to list

searchterms = ["this", "book", "history"]
for string in searchterms:  # iterate through searchterms
    result = wordsafter(string, indata)
    if result:
        print(result)

如何将枚举中的值与关键字匹配？

How to match value in enumeration to a keyword?

python

enumeration