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

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'))