改进用于从名称文件列表中选择名称的代码

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)

我想知道是否有什么方法可以做得更好。

Answer 1

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

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