Улучшение кода для выбора имени из списка имен файлов - PullRequest
0 голосов
/ 13 октября 2018

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

0 голосов
/ 13 октября 2018

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

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