Как красиво прочитать двумерную матрицу?

#python #input #матрицы

Собственно, могу прочитать вот так: https://ideone.com/SCoCuG

def read_lines():
  try:
    line = input()
    while line:
      yield line
      line = input()
  except EOFError:
    pass

def read_matrix():
  return [[int(x) for x in line.split()] for line in read_lines()]

a = read_matrix()
print(a)

b = read_matrix()
print(b)


Но мне кажется, что это как-то не по-питоньи, и должен быть способ красивее?

Входные данные: числа, разделённые пробелами по строке матрицы в каждой строке ввода,
ввод завершается пустой строкой или концом файла.

1 2 3
4 5 6

7 8
9 0
1 2

    

Ответы

Ответ 1

мой вариант https://ideone.com/xwsY1A

import sys
from itertools import takewhile


def read_matrix():
    return [[int(x) for x in l.split()] for l in takewhile(str.strip, sys.stdin)]


a = read_matrix()
print(a)

b = read_matrix()
print(b)