python - 改进从名称文件列表中选择名称的代码

Question

我正在使用 python 中的代码，需要选择一个随机名称。

我有一个包含数千个名称的文本文件（names.txt）。每个名称都在一行中。我想创建一个代码来从此列表中选择一个名称。我已经这样做了：

file_names = open('names.txt', 'r')

names = []
for name in file_names.readlines():
    names.append(name.replace('\n', ''))

random_name = random.choice(names)

我想知道是否有任何方法可以更好地做到这一点。

标签： pythonpython-3.xrandomchoice

Answer 1

您可以改用生成器表达式：

random_name = random.choice(l.strip() for l in open('names.txt', 'r'))