python - 如何在 Python 中选择文件中数字旁边的单词？

Question

假设我打开了这个文件（wordlist.txt）：

...
14613 bore
14614 borg
14615 boric
14616 boris
14621 born
14622 borne
14623 borneo
14624 boron
...

例如，如果变量是 14623，我如何选择它旁边的单词？

Answer 1

如果您只执行一次，请按照 falsetru 显示的方式执行。

如果您需要查找多个数字，请将条目存储在字典中：

wordmap = {}
with open('wordlist.txt') as f:
    for line in f:
        key, value = line.split()
        wordmap[key] = value

现在您可以像这样查找任何数字：

>>> wordmap['14623']
borneo

您可能要添加一些改进：

1. rstrip单词末尾的换行符
2. split(None, 1)处理中间有空格的单词
3. wordmap[int(key)]所以你可以查找wordmap[14623]而不是wordmap['14623']
4. 将整个事情重写为具有理解力的两条线
5. 更好的错误处理，因此如果文件中的一行无效，您可以跳过它，或打印有用的错误消息，如“无效行 #1731：'glassdfefewasd'而不是转储通用异常或任何适当的。

例如，这会执行 #1、#2 和 #4：

with open('wordlist.txt') as f:
    wordmap = dict(line.rstrip().split(None, 1) for line in f)

Answer 2

尝试以下操作：

with open('wordlist.txt') as f:
    for line in f:
        row = line.split()
        if row[0] == '14623':
            print(row[1])
            break

Answer 3

如果您只想访问该项目一次，这是非常可读的：

with open('test.txt') as f:
    print next(line.split()[1] for line in f 
               if line.startswith('14623'))

Answer 4

使用正则表达式：

import re

line_num = 14615
print re.search(r''+str(line_num)+' (\w+)', file_text).group()