定位一行文本中的单词

Question

我在一个文本文件中找到了一行，如下所示：

FLAGS                    = WORD1 WORD2 WORD3

我正在阅读几个文件，其中单词数可以从 0 到最多 3 个不等。

我正在使用此代码：

flag_FLAG = 0
for i in range(len(materialfile)):
    if  "FLAG" in materialfile[i] and "=" in materialfile[i]:
        line_FLAG = i
        flag_FLAG = 1
        
    if flag_FLAG == 1:
        
        temp = materialfile[line_FLAG].split(" ")
        for elem in temp:
            if is_word(elem):
                flags = str(elem)

不幸的是，这样我只能得到一个单词（最后一个）。 “is_word”是我创建的一个函数：

def is_word(s):
    try:
        str(s)
        return True
    except ValueError:
        return False

我想将所有单词作为目标。 我希望我已经说清楚了。

Answer 1

我以这种方式解决：

for i in range(len(materialfile)):
    if  "FLAGS" in materialfile[i] and "="  in materialfile[i]:
        line_flag = i
        flag_flag = 1
if flag_flag == 1:
    flags = materialfile[line_flag].split(" = ")[1].split()

我不知道这是否是一种优雅的方式，但它似乎有效。谢谢

Answer 2

您想要一个嵌套循环，例如：

materialfile = [
    "FLAGS                    = WORD1 WORD2 WORD3",
]

flags = [
    flag
    for line in materialfile if "FLAGS" in line and "=" in line
    for flag in line.split(" = ")[1].split() if flag
]

print(flags)  # ['WORD1', 'WORD2', 'WORD3']

很难说这个确切的代码是否适用于您的实际文件，因为您没有提供示例文件，但希望这能让您指向正确的方向。

请注意，您的 is_word 函数不执行任何操作，因为它们已经是字符串，因此始终会作为无操作转换为 str() 而不会引发异常。上述理解中的 if flag 将过滤掉 flag 中空的值（例如，如果您有像 FLAGS = 这样的行）。