Как читать входные данные, заданные в виде целых чисел, разделенных пробелом и разделенных на разные строки (txt) - PullRequest
0 голосов
/ 10 февраля 2020

Я выполняю эту задачу кодирования в Python, и в этой задаче ввод задается в виде строк целых чисел, разделенных пробелом в файле с именем "input.txt", например:

3 3 1
1 0 3 1
2 2

Каждая строка представляет различную часть ввода, как первая строка - ширина и высота сетки и количество стенок в ней, вторая строка - координаты начала и конца, а третья - координаты указанной c стены. Как бы вы прочитали файл, чтобы в конце вы получили список, где каждая строка представляет собой отдельный элемент в списке, например так:

input = [[3, 3, 1], [1, 0, 3, 1], [2, 2]]
# the numbers are integers

Спасибо за ответ

1 Ответ

0 голосов
/ 10 февраля 2020
with open('input.txt', 'r') as infile:
    # iterating through a file will naturally iterate line-by-line
    # str.split() will split on spaces naturally, and we can convert to int for each value
    # thus, a nested list comprehension
    inp = [[int(i) for i in line.split()] 
           for line in infile]
...