如何根据文本文件中的特定单词过滤特定值并将其存储在列表中？

Question

就像我有一个文本文件 abc.txt 一样，

we 2 rt 3 re 3 tr vh kn mo
we 3 rt 5 re 5 tr yh kn me
we 4 rt 6 re 33 tr ph kn m3
we 5 rt 9 re 34 tr oh kn me
we 6 rt 8 re 32 tr kh kn md

现在我想要针对 tr 的值，过滤后应该得到这个结果，

[vh,yh,ph,oh,kh]

任何人都可以告诉如何做到这一点。应该为其编写什么代码

Answer 1

mylist = [line.split()[7] for line in myfile] 

如果始终是第 8 列，则应该有效。

如果 tr 的位置是可变的，你可以这样做

mylist = []
for line in myfile:
    items = line.split()
    mylist.append(items[items.index("tr")+1])

Answer 2

您可以将行拆分为 before tr 和 after tr 并获取第二部分中的第一个单词。

[ line.split(' tr ')[1].split()[0] for line in file ] 

如果有多个 tr，则表达式将收集第一个之后的单词。或者，这个收集一行中最后一个 tr 之后的单词：

[ line.split(' tr ')[-1].split()[0] for line in file ]

Answer 3

你的问题不太清楚。这就是你所追求的吗？

[line.split()[7] for line in open("abc.txt")]

它返回每行的第八个“单词”。

Answer 4

如果我理解正确，类似这样的事情应该可以完成工作（未经测试）：

resultArray = []
for aString in yourFile:
    anArray = aString.split()
    for i in range(0, len(anArray) - 1):  //-1 in case tr is at the end of array
        if anArray[i] == 'tr':
            resultArray.append(anArray[i + 1])

Answer 5

from operator import itemgetter

# tr value is in the 8th column
tr = itemgetter(7)

print map(tr, (line.split() for line in myfile.readlines()))

Answer 6

使用正则表达式不是更简单吗？

如果 'we' 、 'rt' 、 're' 、 'tr' 在它们的位置上确实是恒定的：

import re

ch = '''
we 2 rt 3 re 3 tr vh kn mo
we 3 rt 5 re 5 tr yh kn me
we 4 rt 6 re 33 tr ph kn m3
we 5 rt 9 re 34 tr oh kn me
we 6 rt 8 re 32 tr kh kn md'''

print re.findall('(?<= tr )([^ ]+)',ch)

如果不是，那么位置就是确定要捕获的内容的标准：

import re

ch = '''
we 2 rt 3 re 3 tr vh kn mo
we 3 rt 5 re 5 tr yh kn me
we 4 rt 6 re 33 tr ph kn m3
we 5 rt 9 re 34 tr oh kn me
we 6 rt 8 re 32 tr kh kn md'''

print [ mat.group(1)
        for mat in re.finditer('^(?:\w+ \d+ ){3}\w+ ([^ ]+) .+',ch,re.M)]

Answer 7

人们可以尝试以下操作：

def filter_words(filename, magic_word):
    with open(filename) as f:
        all_words = f.read().strip().split()
        filtered_words = []
        i = 0
        while True:
            try:
                i = all_words.index(magic_word, i) + 1
                filtered_words.append(all_words[i])
            except IndexError, ValueError:
                break
        return filtered_words

如果“tr”恰好是所提供文本文件中的最后一个单词，则该算法不会失败。

例子：

>>> filter_words('abc.txt', 'tr')
['vh', 'yh', 'ph', 'oh', 'kh']