Улучшение кода для выбора имени из списка имен файлов

Я работаю в коде на 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)

Я хочу знать, есть ли способ сделать это лучше.

1 ответ

Вместо этого вы можете использовать выражение генератора:

random_name = random.choice(l.strip() for l in open('names.txt', 'r'))
Другие вопросы по тегам